【LeetCode】931. 下降路径最小和

931. 下降路径最小和

题解

这题也挺简单的

动态规划

每一行的每一个元素都有可能成为最短路径中的一个元素，因此每行的每个元素都需要遍历

然后第二行中，一个元素作为可能的最短路径元素时，它的上一个元素，肯定是从第一行中它正上方的、左斜方的和右斜方的三个元素中选择一个最小的，因此我们可以得到第二行的每个元素作为最短路径元素时的最短路径值

然后这个值可以作为第三行的上一行，第三行的每个元素作为可能的最短路径元素时，就会从上一行中正上方的、左斜方的、右斜方的三个元素中选择一个最小的，因此我们可以得到第三行的每个元素作为最短路径元素时的最短路径值

以此类推

到最后我们可以得到最后一行的每个元素作为最短路径元素时的最短路径值，然后这些值中的最小值就是最短路径

代码：

    public int minFallingPathSum(int[][] matrix) {
        // 统计一行每个元素被选择时的最短路径数组
        int[] pathArray = new int[matrix.length];
        // 矩阵大小
        int n = matrix.length;

        // 开始遍历矩阵的行
        for (int rowIndex = 0; rowIndex < n; rowIndex++) {
            // 如果是第一行，则初始化最短路径数组
            if (rowIndex == 0) {
                for (int colIndex = 0; colIndex < n; colIndex++) {
                    pathArray[colIndex] = matrix[0][colIndex];
                }
                continue;
            }

            int[] tempPathArray = new int[matrix.length];
            // 不是第一行时，遍历这一行的每个元素
            for (int colIndex = 0; colIndex < n; colIndex++) {
                // 找到这个元素作为路径时，它的上一步应该选哪个会最小
                int center = pathArray[colIndex];
                int left = (colIndex != 0) ? pathArray[colIndex - 1] : 100 * rowIndex;
                int right = (colIndex != n - 1) ? pathArray[colIndex + 1] : 100 * rowIndex;
                tempPathArray[colIndex] = Math.min(Math.min(center, left), right) + matrix[rowIndex][colIndex];
            }
            // 更新最短路径数组
            pathArray = Arrays.copyOf(tempPathArray, tempPathArray.length);
        }

        // 遍历最短路径数组，找最短路径
        return Arrays.stream(pathArray).min().getAsInt();
    }

Java知识UP

• 数组如何复制（拷贝）

  参考链接

  有四种方式，我用的是Arrays类的静态方法copyOf，参数是需要拷贝的数组，以及你想要拷贝这个数组多长的元素

• 如何找出一个数组的最小值

  参考链接

  有三种方式，我用的是把数组放进stream里面（Arrays.stream(array)），然后调用这个对象的min方法（当然也有对应的max方法），最后这个方法的结果还需要再调用这个对象的getAsInt方法转成int类型

  当然也可以直接调用Arrays类型的sort方法先排序，然后根据需求获取元素的第一个或者最后一个元素

  stream用起来很方便