列表

详情

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 中只有一个元素，数组的三角和为这个元素自己。

 

提示：

原站题解

去查看

上次编辑到这里，代码来自缓存 点击恢复默认模板
class Solution { public: int triangularSum(vector<int>& nums) { } };

golang 解法, 执行用时: 44 ms, 内存消耗: 4.8 MB, 提交时间: 2022-11-10 20:02:32 

func triangularSum(a []int) int {
	for n := len(a) - 1; n > 0; n-- {
		for i := 0; i < n; i++ {
			a[i] = (a[i] + a[i+1]) % 10
		}
	}
	return a[0]
}

python3 解法, 执行用时: 2936 ms, 内存消耗: 15.1 MB, 提交时间: 2022-11-10 19:57:51 

class Solution:
    def triangularSum(self, nums: List[int]) -> int:
        n = len(nums)
        if n < 2: return nums[0]
        for i in range(n-1):
            nums = [(num + nums[i+1])%10 for i, num in enumerate(nums) if i < len(nums)-1]
        return nums[0]

上一题

© 2024 leetcode.cn