leetcode-数组的三角和

94 阅读1分钟

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

题目描述

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

2221示例1.png

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

示例 2:

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

提示:

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

思路

今天的每日一题没做出来,换个随机一题水一水。
本题类似于杨辉三角,可以直接模拟,一层一层往下计算直到只剩下1个数字。一开始有length个数,就需要计算length-1层,每次需要计算的次数为length - 1 - i。一个小技巧是,nums[j]和nums[j+1]的和可以直接更新到nums[j],因为当前一次循环中,nums[j]不会再被使用到了,这样不需要额外的存储空间,其实也类似于有些动态规划的题目减少使用额外空间的技巧。

Java版本代码

class Solution {
    public int triangularSum(int[] nums) {
        for (int i = 0; i < nums.length - 1; i++) {
            for (int j = 0; j < nums.length - 1 - i; j++) {
                nums[j] += nums[j+1];
                nums[j] %= 10;
            }
        }
        return nums[0];
    }
}