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]
}
|