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;
}
};