leetCode-2221: 数组的三角和

题目描述

给你一个下标从 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）双层循环暴力去解，或者加上动态规划也可以解答；
（2）循环时需要注意内层 j 的循环从 0 开始，到 nums.length - i 结束，因为每次循环都较上次少一个数；
（3）最后返回数组最后最后一行第 1 个元素即可。

代码展示

public class TriangularSum {

    public int triangularSum(int[] nums) {
        int[][] dp = new int[nums.length][nums.length];
        dp[0] = nums;
        for (int i = 1; i < nums.length; i++) {
            for (int j = 0; j < nums.length - i; j++) {
                dp[i][j] = (dp[i - 1][j] + dp[i - 1][j + 1]) % 10;
            }
        }
        // 将二维数组打印
        display(dp);
        return dp[nums.length - 1][0];
    }

    private void display(int[][] dp) {
        for (int i = 0; i < dp.length; i++) {
            for (int j = 0; j < dp[i].length; j++) {
                System.out.printf("%4d", dp[i][j]);
            }
            System.out.println();
        }
    }

    public static void main(String[] args) {
        int[] nums = {1, 2, 3, 4, 5};
        System.out.println(new TriangularSum().triangularSum(nums));
    }

}