题目表明去终点只能往下和往右,回去只能往上和往左,统计路程中经过的樱桃个数,那么就相当于从起点出发两次到终点,但两次路过到的同个樱桃就不能再重复计算,这两次不分先后,同时走深搜,然后用dp表示每个状态的最优解。
dp[x1][y1][x2]代表同时走两条路分别到(x1,y1)(x2,y2)的话,继续往下走所得到的最大值。因为是同时,他们两条路从起点开始移动的曼哈顿距离相等,dp有以下特点:
- 距离相等,就有x1+y1 == x2+y2,则y2=x1+y1-x2,所以可以用三维代表一个状态
- 遍历从起点开始,dp初始化为INT_MIN表示未经过,-1表示已表示不可达
- 未经过和不可达的意义是不同的,不要混淆
- 【越界】或【已经过且确定不可达的点(两条路的状态)】直接返回-1,表示不可达
- 已经过的dp也直接返回该状态的最优解
- 深搜的情况只有一路下另一路下、一路下另一路右、一路右另一路下、一路右另一路右,四种情况
- 记录四种情况并找到最优解,然后判断当前状态的两个位置是否有樱桃
- 特别判断两个位置是否重合,只有不重合才能算两个不同的位置
出于个人习惯,本代码用了vector作为dp,写完发现vector这样写似乎并不好,(觉得vector这样看起来写太怪了,但能用就行),但现在觉得用普通的三维数组、初始化用memset更好。
附上C++代码:
class Solution {
public:
int dfs(vector<vector<int>>& grid,vector<vector<vector<int>>> &dp,int x1,int y1,int x2){
int y2 = x1+y1-x2;
if(x1 == row||y1 == col || x2 == row || y2==col||grid[x1][y1] == -1||grid[x2][y2] == -1)
return -1;
else if(dp[x1][y1][x2]!= INT_MIN)
return dp[x1][y1][x2];
else if(x1 == row-1&&y1 == col-1)
return grid[x1][y1];
int k = dfs(grid,dp,x1,y1+1,x2);
k = max(k,dfs(grid,dp,x1,y1+1,x2+1));
k = max(k,dfs(grid,dp,x1+1,y1,x2));
k = max(k,dfs(grid,dp,x1+1,y1,x2+1));
if(k>=0){
k += grid[x1][y1];
if(x1!=x2&&y1!=y2)k+=grid[x2][y2];
}
dp[x1][y1][x2] = k;
return k;
}
int cherryPickup(vector<vector<int>>& grid) {
row = grid.size();
col = grid[0].size();
vector<vector<vector<int>>> dp(row,vector<vector<int>>(col,vector<int>(row,INT_MIN)));
return max(0,dfs(grid,dp,0,0,0));
}
};