LeetCode探索（125）：2221-数组的三角和

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第22天，点击查看活动详情 >>

题目

给你一个下标从 0 开始的整数数组 nums ，其中 nums[i] 是 0 到 9 之间（两者都包含）的一个数字。

nums 的 三角和 是执行以下操作以后最后剩下元素的值：

1. nums 初始包含 n 个元素。如果 n == 1 ，终止 操作。否则，创建 一个新的下标从 0 开始的长度为 n - 1 的整数数组 newNums 。
2. 对于满足 0 <= i < n - 1 的下标 i ，newNums[i] 赋值 为 (nums[i] + nums[i+1]) % 10 ，% 表示取余运算。
3. 将 newNums 替换 数组 nums 。
4. 从步骤 1 开始 重复 整个过程。

请你返回 nums 的三角和。

示例 1：

输入：nums = [1,2,3,4,5]
输出：8
解释：
上图展示了得到数组三角和的过程。

示例 2：

输入：nums = [5]
输出：5
解释：
由于 nums 中只有一个元素，数组的三角和为这个元素自己。

提示：

• 1 <= nums.length <= 1000
• 0 <= nums[i] <= 9

思考

本题难度中等。

首先是读懂题意。题目中给出的整数数组 nums，满足数组长度 >= 1，且 0 <= nums[i] <= 9。每次操作，对数组中每组相邻的两个元素求和，最后替换该数组。该操作重复至数组长度为1为止。

我们可以模拟这个操作流程。当数组长度 > 1时，对数组中每组相邻的两个元素求和，得到的新数组替换原来的数组，并继续进行操作。如果数组长度为1，直接返回数组的第一个元素即可。

解答

方法一：模拟

/**
 * @param {number[]} nums
 * @return {number}
 */
var triangularSum = function(nums) {
  while (nums.length > 1) {
    let arr = []
    for (let i = 0; i < nums.length - 1; i++) {
      arr[i] = (nums[i] + nums[i + 1]) % 10
    }
    nums = arr
  }
  return nums[0]
}
// 执行用时：196 ms, 在所有 JavaScript 提交中击败了19.79%的用户
// 内存消耗：47.9 MB, 在所有 JavaScript 提交中击败了19.79%的用户
// 通过测试用例：300 / 300

复杂度分析：

• 时间复杂度：O(n^2)。
• 空间复杂度：O(n)。

参考

• 数组的三角和