LeetCode 1463. 摘樱桃 II（dp）

题意：

给你一个 rows x cols 的矩阵 grid 来表示一块樱桃地。 grid 中每个格子的数字表示你能获得的樱桃数目。

你有两个机器人帮你收集樱桃，机器人 1 从左上角格子 (0,0) 出发，机器人 2 从右上角格子 (0, cols-1) 出发。

请你按照如下规则，返回两个机器人能收集的最多樱桃数目：

从格子 (i,j) 出发，机器人可以移动到格子 (i+1, j-1)，(i+1, j) 或者 (i+1, j+1) 。
当一个机器人经过某个格子时，它会把该格子内所有的樱桃都摘走，然后这个位置会变成空格子，即没有樱桃的格子。
当两个机器人同时到达同一个格子时，它们中只有一个可以摘到樱桃。
两个机器人在任意时刻都不能移动到 grid 外面。
两个机器人最后都要到达 grid 最底下一行。

数据范围:
rows == grid.length
cols == grid[i].length
2 <= rows, cols <= 70
0 <= grid[i][j] <= 100 

解法：

显然令d[i][j][k]表示第i行,机器人1在位置j,机器人在位置k能获得的最大值,
枚举机器人走的方向进行dp即可.

code：

int d[71][71][71];
class Solution {
public:
    int cherryPickup(vector<vector<int>>& a) {
        memset(d,-1,sizeof d);
        int n=a.size();
        int m=a[0].size();
        d[0][0][m-1]=a[0][0]+a[0][m-1];
        for(int i=0;i<n-1;i++){
            for(int j=0;j<m;j++){
                for(int k=0;k<m;k++){//枚举当前状态d[i][j][k]
                    if(d[i][j][k]==-1)continue;//状态不合法
                    for(int d1=-1;d1<=1;d1++){
                        for(int d2=-1;d2<=1;d2++){//枚举两个机器人走的方向
                            int x1=j+d1;
                            int x2=k+d2;
                            if(x1<0||x1>=m||x2<0||x2>=m)continue;
                            int temp=a[i+1][x1];
                            if(x2!=x1)temp+=a[i+1][x2];
                            d[i+1][x1][x2]=max(d[i+1][x1][x2],d[i][j][k]+temp);
                        }
                    }
                }
            }
        }
        int ans=0;
        for(int j=0;j<m;j++){
            for(int k=0;k<m;k++){
                ans=max(ans,d[n-1][j][k]);
            }
        }
        return ans;
    }
};

---