题目:
给你一个整数数组 arr ,你一开始在数组的第一个元素处(下标为 0)。
每一步,你可以从下标 i 跳到下标:
i + 1 满足:i + 1 < arr.length
i - 1 满足:i - 1 >= 0
j 满足:arr[i] == arr[j] 且 i != j
请你返回到达数组最后一个元素的下标处所需的 最少操作次数 。
注意:任何时候你都不能跳到数组外面。
解答:
class Solution:
def minJumps(self, arr: List[int]) -> int:
n=len(arr)
if n==1:
return 0
#dic[i]:统计与arr[i]相等的元素的下标
dic=defaultdict(list)
for i,val in enumerate(arr):
dic[val].append(i)
dist=[inf]*n
dist[0]=0
def visited(idx,step):
#若当前索引未被访问
if dist[idx]==inf:
d.append(idx)
dist[idx]=step
d=deque([0])
while len(d)>0:
t=d.popleft()
step=dist[t]
if t==n-1:
return step
#【等值跳】
for idx in dic[arr[t]]:
#若当前元素未被访问过
visited(idx,step+1)
#为避免重复访问,将当前值移除
del dic[arr[t]]
#后跳
visited(t+1,step+1)
#前跳
if t!=0:
visited(t-1,step+1)