2221. 数组的三角和

题目

给你一个下标从 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

解题思路

整个过程可以用一个大的循环来控制,循环的条件是数组(在 Go 中我们用切片 slice 更方便)的长度大于 1。在每一次循环中,我们都生成一个新的、长度减一的切片,这个切片可以原地生成。

具体代码

func triangularSum(nums []int) int {

    n := len(nums)
    for i := n - 1; i >= 0; i-- {
        s := nums[:i + 1]
        for j := 0; j < i; j++ {
            s[j] = (s[j] + s[j + 1]) % 10
        }
    }

    return nums[0]  
}