• 最大路径和问题(摘樱桃问题)


    最大路径和问题(摘樱桃问题)

    作者:Grey

    原文地址: 最大路径和问题(摘樱桃问题)

    题目链接

    LeetCode 741. 摘樱桃

    主要思路

    本题的难点在于尝试,如何模拟一来一回的情况,我们可以这样做,定义两个小人,两个人都从(0,0)位置出发,到右下角位置,每人同时选择不同的下一步,如果两个小人跳到了同一个位置,只计算一份。由于两个小人是同时走,而且每次只走一步,所以,两个小人一定是同时到达右下角的,两个小人一路收集的樱桃数量,就是一来一回收集的数量。

    我们可以写出第一个尝试版本,定义递归函数

    int p(int[][] matrix, int m, int n, int x1, int y1, int x2, int y2)
    
    • 1

    递归函数的含义表示,第一个小人从(x1,y1)开始到右下角,第二个小人从(x2,y2)开始到右下角,获取到的最大值是多少。所以p(matrix, m, n, 0, 0, 0, 0)就是我们需要的答案。

    接下来考虑base case,即:两个小人都到达了右下角,由于题目已经说了,右下角的值一定不是-1,所以,可以获得一份樱桃数量。

            if (x1 == m - 1 && y1 == n - 1) {
                // 已经到了最右下角了
                // 隐含条件:另外一个点也一定到达右下角
                // 获得一份樱桃数
                return matrix[x1][y1];
            }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6

    接下来就是普遍尝试,分别可以分成如下四种情况:

    情况1:第一个小人往下,第二个小人往右。

    情况2:第一个小人往下,第二个小人往下。

    情况3:第一个小人往右,第二个小人往下。

    情况4:第一个小人往右,第二个小人往右。

    但是在走上述任何分支的时候,记得要满足两个条件

    第一个条件,不能越界

    第二个条件,下一个要走的位置不能是-1,因为题目说到,-1是不能走的位置。

            int next = -1;
            // 下,下
            if (x1 + 1 < m && x2 + 1 < m && matrix[x1 + 1][y1] != -1 && matrix[x2 + 1][y2] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2 + 1, y2), next);
            }
            // 下,右
            if (x1 + 1 < m && y2 + 1 < n && matrix[x1 + 1][y1] != -1 && matrix[x2][y2 + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2, y2 + 1), next);
            }
            // 右,下
            if (y1 + 1 < n && x2 + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2 + 1][y2] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2 + 1, y2), next);
            }
            // 右,右
            if (y1 + 1 < n && y2 + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2][y2 + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2, y2 + 1), next);
            }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17

    经过上述四个分支,如果next的值还是-1,说明无路可走,返回-1

            if (next == -1) {
                return -1;
            }
    
    • 1
    • 2
    • 3

    如果next不等于-1,说明有可以走的路径,那么继续判断两个小人是否在同一位置,如果在同一个位置,则只收集一份樱桃,如果不在同一个位置,收集两个小人所在位置的樱桃之和。

            if (x1 == x2) {
                // 到达同一个位置,只取一个值
                return next + matrix[x1][y1];
            }
            return next + matrix[x1][y1] + matrix[x2][y2];
    
    • 1
    • 2
    • 3
    • 4
    • 5

    注:判断同一个位置,只需要一个维度上的坐标相等即可

    完整代码

        public static int cherryPickup1(int[][] matrix) {
            return Math.max(0, p(matrix, matrix.length, matrix[0].length, 0, 0, 0, 0));
        }
    
        // 定义两个小人,两个人都从(0,0)位置出发,到右下角位置,每人同时选择不同的下一步,如果两个小人跳到了同一个位置,只计算一份。
        public static int p(int[][] matrix, int m, int n, int x1, int y1, int x2, int y2) {
            if (x1 == m - 1 && y1 == n - 1) {
                // 已经到了最右下角了
                // 隐含条件:另外一个点也一定到达右下角
                // 获得一份樱桃数
                return matrix[x1][y1];
            }
    
            int next = -1;
            // 下,下
            if (x1 + 1 < m && x2 + 1 < m && matrix[x1 + 1][y1] != -1 && matrix[x2 + 1][y2] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2 + 1, y2), next);
            }
            // 下,右
            if (x1 + 1 < m && y2 + 1 < n && matrix[x1 + 1][y1] != -1 && matrix[x2][y2 + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2, y2 + 1), next);
            }
            // 右,下
            if (y1 + 1 < n && x2 + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2 + 1][y2] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2 + 1, y2), next);
            }
            // 右,右
            if (y1 + 1 < n && y2 + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2][y2 + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2, y2 + 1), next);
            }
            if (next == -1) {
                return -1;
            }
            if (x1 == x2) {
                // 到达同一个位置,只取一个值
                return next + matrix[x1][y1];
            }
            return next + matrix[x1][y1] + matrix[x2][y2];
        }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38
    • 39

    这个解法在 LeetCode 上直接超时。

    在上述尝试的基础上,我们可以做进一步的优化,递归函数现在是四个可变参数,根据我们的设计,其实可以得到如下公式

    x1 + y1 = x2 + y2
    
    • 1

    那么我们可以将递归函数省略一个参数y2,因为

    y2 = x1 + y1 - x2
    
    • 1

    上述递归方法我们可以修改为

    // 省略y2参数
        public static int p(int[][] matrix, int m, int n, int x1, int y1, int x2) {
            if (x1 == m - 1 && y1 == n - 1) {
                // 已经到了最右下角了
                // 隐含条件:另外一个点也一定到达右下角
                // 获得一份樱桃数
                return matrix[x1][y1];
            }
    
            int next = -1;
            // 下,下
            if (x1 + 1 < m && x2 + 1 < m && matrix[x1 + 1][y1] != -1 && matrix[x2 + 1][getY2(x1, y1, x2)] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2 + 1), next);
            }
            // 下,右
            if (x1 + 1 < m && getY2(x1, y1, x2) + 1 < n && matrix[x1 + 1][y1] != -1 && matrix[x2][getY2(x1, y1, x2) + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2), next);
            }
            // 右,下
            if (y1 + 1 < n && x2 + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2 + 1][getY2(x1, y1, x2)] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2 + 1), next);
            }
            // 右,右
            if (y1 + 1 < n && getY2(x1, y1, x2) + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2][getY2(x1, y1, x2) + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2), next);
            }
            if (next == -1) {
                return -1;
            }
            if (x1 == x2) {
                // 到达同一个位置,只取一个值
                return next + matrix[x1][y1];
            }
            return next + matrix[x1][y1] + matrix[x2][getY2(x1, y1, x2)];
        }
        public static int getY2(int x1, int y1, int x2) {
            return x1 + y1 - x2;
        }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38

    我们可以做进一步的优化,由于三个可变参数的范围分别是[0...m-1][0...n-1][0...m-1],那么我们可以设置三维dp,将所有递归结果缓存起来

    int[][][] dp = new int[m][n][m];
    
    • 1

    三维dp的所有位置初始值设置为Integer.MIN_VALUE,在递归方法中,将这个三维表作为缓存带入参数中,每次递归过程中,先从dp表中拿值,如果值不为Integer.MIN_VALUE,说明这个过程算过,直接取值即可。

        public static int p(int[][] matrix, int m, int n, int x1, int y1, int x2, int[][][] dp) {
            if (dp[x1][y1][x2] != Integer.MIN_VALUE) {
                // 说明这个过程曾经算过,直接从缓存中取值即可
                return dp[x1][y1][x2];
            }
        }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6

    然后在每次递归过程中,将答案存入缓存,完整代码如下

        // 动态规划
        // 三维表
        public static int cherryPickup(int[][] matrix) {
            int m = matrix.length;
            int n = matrix[0].length;
            int[][][] dp = new int[m][n][m];
            for (int i = 0; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    for (int k = 0; k < m; k++) {
                        // 初始值设置为Integer.MIN_VALUE,只要不等于Integer.MIN_VALUE,就说明递归过程已经算过了
                        dp[i][j][k] = Integer.MIN_VALUE;
                    }
                }
            }
            return Math.max(0, p(matrix, m, n, 0, 0, 0, dp));
        }
    
        // 定义两个小人,两个人都从(0,0)位置出发,到右下角位置,每人同时选择不同的下一步,如果两个小人跳到了同一个位置,只计算一份。
        public static int p(int[][] matrix, int m, int n, int x1, int y1, int x2, int[][][] dp) {
            if (dp[x1][y1][x2] != Integer.MIN_VALUE) {
                // 说明这个过程曾经算过,直接从缓存中取值即可
                return dp[x1][y1][x2];
            }
            if (x1 == m - 1 && y1 == n - 1) {
                // 已经到了最右下角了
                // 隐含条件:另外一个点也一定到达右下角
                // 获得一份樱桃数
                dp[x1][y1][x2] = matrix[x1][y1];
                return matrix[x1][y1];
            }
    
            int next = -1;
            // 下,下
            if (x1 + 1 < m && x2 + 1 < m && matrix[x1 + 1][y1] != -1 && matrix[x2 + 1][getY2(x1, y1, x2)] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2 + 1, dp), next);
            }
            // 下,右
            if (x1 + 1 < m && getY2(x1, y1, x2) + 1 < n && matrix[x1 + 1][y1] != -1 && matrix[x2][getY2(x1, y1, x2) + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1 + 1, y1, x2, dp), next);
            }
            // 右,下
            if (y1 + 1 < n && x2 + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2 + 1][getY2(x1, y1, x2)] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2 + 1, dp), next);
            }
            // 右,右
            if (y1 + 1 < n && getY2(x1, y1, x2) + 1 < m && matrix[x1][y1 + 1] != -1 && matrix[x2][getY2(x1, y1, x2) + 1] != -1) {
                next = Math.max(p(matrix, m, n, x1, y1 + 1, x2, dp), next);
            }
            if (next == -1) {
                dp[x1][y1][x2] = -1;
                return -1;
            }
            if (x1 == x2) {
                // 到达同一个位置,只取一个值
                // 将答案存入缓存
                dp[x1][y1][x2] = next + matrix[x1][y1];
                return dp[x1][y1][x2];
            }
            // 将答案存入缓存
            dp[x1][y1][x2] = next + matrix[x1][y1] + matrix[x2][getY2(x1, y1, x2)];
            return dp[x1][y1][x2];
        }
    
        public static int getY2(int x1, int y1, int x2) {
            return x1 + y1 - x2;
        }
    
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9
    • 10
    • 11
    • 12
    • 13
    • 14
    • 15
    • 16
    • 17
    • 18
    • 19
    • 20
    • 21
    • 22
    • 23
    • 24
    • 25
    • 26
    • 27
    • 28
    • 29
    • 30
    • 31
    • 32
    • 33
    • 34
    • 35
    • 36
    • 37
    • 38
    • 39
    • 40
    • 41
    • 42
    • 43
    • 44
    • 45
    • 46
    • 47
    • 48
    • 49
    • 50
    • 51
    • 52
    • 53
    • 54
    • 55
    • 56
    • 57
    • 58
    • 59
    • 60
    • 61
    • 62
    • 63
    • 64
    • 65
    • 66

    时间复杂度O(m*n*m),LeetCode 中直接AC。

    更多

    算法和数据结构笔记

  • 相关阅读:
    完蛋!百融云被大阳线包围了!
    node+mysql+navicat一条龙流程
    【无人机】基于Matlab模拟无人机群跟踪固定目标
    各种 Dice Loss 变体
    任何样式,javascript都可以操作,让你所向披靡
    【探索Linux】—— 强大的命令行工具 P.23(线程池 —— 简单模拟)
    linux下vxlan理论学习及实战配置方法
    医院项目-预约挂号-第二部分-前端
    微信浏览器自动播放音频(兼容Android和iOS)
    [深度学习] 大模型学习8上-推理部署框架llama.cpp与Ollama使用指北
  • 原文地址:https://blog.csdn.net/hotonyhui/article/details/125479933