[leetcode]741摘樱桃,动态规划

题目表明去终点只能往下和往右,回去只能往上和往左,统计路程中经过的樱桃个数,那么就相当于从起点出发两次到终点,但两次路过到的同个樱桃就不能再重复计算,这两次不分先后,同时走深搜,然后用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));
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值