leetcode 403.青蛙过河 - 动态规划

leetcode 403.青蛙过河 - 动态规划

题干

一只青蛙想要过河。 假定河流被等分为若干个单元格，并且在每一个单元格内都有可能放有一块石子（也有可能没有）。 青蛙可以跳上石子，但是不可以跳入水中。
给你石子的位置列表 stones（用单元格序号 升序 表示）， 请判定青蛙能否成功过河（即能否在最后一步跳至最后一块石子上）。
开始时， 青蛙默认已站在第一块石子上，并可以假定它第一步只能跳跃一个单位（即只能从单元格 1 跳至单元格 2 ）。
如果青蛙上一步跳跃了 k 个单位，那么它接下来的跳跃距离只能选择为 k - 1、k 或 k + 1 个单位。 另请注意，青蛙只能向前方（终点的方向）跳跃。

示例 1：
输入：stones = [0,1,3,5,6,8,12,17]
输出：true
解释：青蛙可以成功过河，按照如下方案跳跃：跳 1 个单位到第 2 块石子, 然后跳 2 个单位到第 3 块石子, 接着 跳 2 个单位到第 4 块石子, 然后跳 3 个单位到第 6 块石子, 跳 4 个单位到第 7 块石子, 最后，跳 5 个单位到第 8 个石子（即最后一块石子）。

示例 2：
输入：stones = [0,1,2,3,4,8,9,11]
输出：false
解释：这是因为第 5 和第 6 个石子之间的间距太大，没有可选的方案供青蛙跳跃过去。

提示：
2 <= stones.length <= 2000
0 <= stones[i] <= 231 - 1
stones[0] == 0

知识点&算法

不能过的bfs

看到这个思路很容易想到用bfs模拟，然后果不其然直接超时（甚至只过了15/50）

class Solution {
public:
    bool canCross(vector<int>& stones) {
        int n = stones.size();
        queue<pair<vector<int>::iterator,int> > Q;
        for(int i = 1 ; i < n ; ++i){
            if(stones[i] == 1){
                Q.push({stones.begin() + i,1});
                if(i == n - 1) return true;
            }
        }
        while(!Q.empty()){
            auto now = Q.front();
            Q.pop();
            if(now.first == stones.end() - 1) return true;
            for(int diff = -1 ; diff <= 1 ; ++diff){
                int target = *now.first + now.second + diff;
                auto itr = lower_bound(now.first + 1,stones.end(),target);
                if(itr != stones.end() && *itr == target){
                    Q.push({itr,now.second + diff});
                }
            }
        }   
        return false;
    }
};

其实记忆化搜索可以过，但是bfs不像dfs那样有回溯的特性，所以优化比较难实现…

能过的动态规划

毕竟很多暴搜的题目都能用动规做，既然暴搜搜不出来，就试试动规
用一个二维数组dp[i][j]来维护到达下标为i的石头，跳数为j的状态的可能性，由于跳数只能为前一跳跳数 + -1,0,1所以很容易得到状态转移方程：
对于石头i左侧的一石头k来说
dp[i][j] = dp[k][j-1] || dp[k][j] || dp[k][j+1];
只要任何dp[n-1][j]为true，即说明存在跳到最后石头的可能性。
考虑边界条件：
dp[0][0]表示在下标为0的石头上（即初始点），因为第一跳跳数固定为1，所以dp[0][0]置true即可。
对于每个石头都要遍历其左侧的全部石头。

题解

class Solution {
public:
    bool canCross(vector<int>& stones) {
        int n = stones.size();
        bool dp[n+5][n+5];
        memset(dp,false,sizeof(dp) );
        dp[0][0] = true;
        for(int i = 1 ; i < n ; ++i){
            for(int j = i - 1 ; j >= 0 ; --j){
                int dis = stones[i] - stones[j];
				//在极限情况下，即每次跳数+1，且跳到相邻石头的情况下，跳到下标为i的石头时的最大跳数为i。
				//所以如果要求跳数大于了可能跳数的最大情况，直接break即可
                if(dis > j + 1) break;
                dp[i][dis] = dp[j][dis-1] || dp[j][dis] || dp[j][dis+1];
                if(i == n - 1 && dp[i][dis]) return true;
            }
        }
        return false;
    }
};