1. 问题描述:
一只青蛙想要过河。 假定河流被等分为若干个单元格,并且在每一个单元格内都有可能放有一块石子(也有可能没有)。 青蛙可以跳上石子,但是不可以跳入水中。给你石子的位置列表 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] <= 2 ^ 31 - 1
stones[0] == 0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/frog-jump
2. 思路分析:
因为我们无法确定每一次怎么样走才可以从起点到达终点,所以需要枚举所有的走法,也即可以使用递归求解(递归可以求解所有的可能性)。分析题目可以知道我们可以从起点出发,依次尝试三种可能的走法(对应递归的三种平行状态),在递归的方法中需要传递两个参数,第一个是当前青蛙的位置p,第二个是到达当前位置的速度k,传递到达当前位置的速度是因为从当前位置往下走的速度与到达当前位置的速度是相关的,但是如果使用这样的方法递归会存在很多子问题的重复求解,因为递归方法中存在两个动态变化的参数,为当前青蛙的位置和到达当前位置的速度,在递归的时候很可能是之前求解过这样的状态了,所以当递归到达同样状态的时候会与之前一样递归所以就存在子问题的重复求解了。所以我们需要使用记忆型的递归(记忆型的递归一般是根据递归方法中动态变化的参数的个数来决定使用多少维的数据结构来记录求解过的状态,在递归的时候记录,递归前判断是否存在已经求解过),记录下之前求解过的状态,也即上面对应的两个动态变化的参数,因为使用的是python语言所以我们可以字典来记录,键为元组对应着青蛙的位置和到达当前位置的速度,在往下递归之前先判断之前是否已经求解过当前的状态,如果求解过那么直接返回False,因为这个状态如果合法说明之前就已经返回True了。我们在一开始的时候可以使用字典来记录石子的位置,这样在往下递归的时候才可以判断下一个位置是否存在石子,如果没有石子说明是不合法的,返回False,当我们发现青蛙的位置到达了最后一个位置的时候直接返回True,因为只要是能到达终点说明是合法的,在递归的时候尝试三种平行的状态,也即对应三种不同的速度到达下一个位置。
3. 代码如下:
from typing import List
class Solution:
# 递归的第一个参数p表示青蛙当前的位置, 第二个参数k表示跳到当前位置的速度, 使用哈希表来记录已经求解过的状态, 对应这两个动态变化的参数这样可以避免子问题的重复求解
def dfs(self, p: int, k: int, dic: dict, des: int, stones_pos: dict):
# 青蛙的位置超过了最后一个位置或者下一个位置没有石头返回False
if p < 0 or p > des or p not in stones_pos: return False
# 当前的状态之前已经求解过
if (p, k) in dic: return False
# 青蛙到达了最终的位置
if p == des: return True
# 记录求解过的状态
dic[(p, k)] = 1
# 尝试三种平行状态, 主要是有一个为True那么就会层层返回True
return self.dfs(p + k - 1, k - 1, dic, des, stones_pos) or self.dfs(p + k, k, dic, des, stones_pos) or self.dfs(
p + k + 1, k + 1, dic, des, stones_pos)
def canCross(self, stones: List[int]) -> bool:
if not stones: return False
# 使用字典来记录石头的位置
stones_pos = dict()
for x in stones:
stones_pos[x] = 1
dic = dict()
return self.dfs(stones[0], 0, dic, stones[-1], stones_pos)