1. 问题描述:
给你一个下标从 0 开始的整数数组 nums ,其中 nums[i] 是 0 到 9 之间(两者都包含)的一个数字。nums 的三角和是执行以下操作以后最后剩下元素的值:nums 初始包含 n 个元素。如果 n == 1 ,终止操作。否则,创建 一个新的下标从 0 开始的长度为 n - 1 的整数数组 newNums 。对于满足 0 <= i < n - 1 的下标 i ,newNums[i] 赋值为 (nums[i] + nums[i+1]) % 10 ,% 表示取余运算。将 newNums 替换数组 nums 。从步骤 1 开始重复整个过程。请你返回 nums 的三角和。
示例 1:
输入:nums = [1,2,3,4,5]
输出:8
解释:
上图展示了得到数组三角和的过程。
示例 2:
输入:nums = [5]
输出:5
解释:
由于 nums 中只有一个元素,数组的三角和为这个元素自己。
提示:
1 <= nums.length <= 1000
0 <= nums[i] <= 9
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-triangular-sum-of-an-array/
2. 思路分析:
分析题目可以知道我们可以模拟相加的过程,每一次操作得到一个新的newNums,当执行n - 1次之后newNums的第一个元素就是答案,所以模拟整个过程即可,并且进一步可以发现其实整个过程是杨辉三角逆推的过程,nums 是杨辉三角的最后一行的元素,我们可以通过递推得到倒数第二行的值...一直递推到第一行,最终第一行的值就是答案。
3. 代码如下:
python:
from typing import List
class Solution:
def triangularSum(self, nums: List[int]) -> int:
# 从最后一行开始递推
for n in range(len(nums) - 1, 0, -1):
for i in range(n):
nums[i] = (nums[i] + nums[i + 1]) % 10
return nums[0]
go:
package main
import "fmt"
// 模拟相加的过程得到newNums
func get(slice []int) []int {
res := []int{}
for i := 1; i < len(slice); i++ {
// 通过append方法向切片res的末尾添加一个元素
res = append(res, (slice[i]+slice[i-1])%10)
}
return res
}
func triangularSum(nums []int) int {
// slice为切片
slice := []int{}
// 将nums中的元素添加到slice
slice = append(slice, nums...)
for i := 0; i < len(nums)-1; i++ {
// t记录当前操作之后得到的切片结果
t := []int{}
t = append(t, slice...)
// 更新下一次需要操作的slice
slice = get(t)
}
return slice[0]
}
package main
import "fmt"
func triangularSum(nums []int) int {
// 可以看成是杨辉三角的逆推
for n := len(nums) - 1; n > 0; n-- {
for i := 0; i < n; i++ {
nums[i] = (nums[i] + nums[i+1]) % 10
}
}
return nums[0]
}