leetcode 1345. 跳跃游戏 IV(C++)
给你一个整数数组 arr ,你一开始在数组的第一个元素处(下标为 0)。
每一步,你可以从下标 i 跳到下标:
-
i + 1 满足:i + 1 < arr.length i - 1 满足:i - 1 >= 0 j 满足:arr[i] == arr[j] 且 i != j
请你返回到达数组最后一个元素的下标处所需的 最少操作次数 。
注意:任何时候你都不能跳到数组外面。
示例 1:
输入:arr = [100,-23,-23,404,100,23,23,23,3,404] 输出:3 解释:那你需要跳跃 3 次,下标依次为 0 --> 4 --> 3 --> 9 。下标 9 为数组的最后一个元素的下标。
示例 2:
输入:arr = [7] 输出:0 解释:一开始就在最后一个元素处,所以你不需要跳跃。
示例 3:
输入:arr = [7,6,9,6,9,6,9,7] 输出:1 解释:你可以直接从下标 0 处跳到下标 7 处,也就是数组的最后一个元素处。
示例 4:
输入:arr = [6,1,9] 输出:2
示例 5:
输入:arr = [11,22,7,7,7,7,7,7,7,22,13] 输出:3
提示:
-
1 <= arr.length <= 5 * 10^4 -10^8 <= arr[i] <= 10^8
C++
class Solution { public: int minJumps(vector<int>& arr) { int n=arr.size(); vector<int> res(n,50000); map<int,vector<int>> tmp; for(int i=0;i<n;i++) { tmp[arr[i]].push_back(i); } queue<int> que; que.push(0); res[0]=0; while(!que.empty()) { int i=que.front(); que.pop(); if(i-1>=0 && i-1<n) { if(res[i-1]>res[i]+1) { res[i-1]=res[i]+1; que.push(i-1); } } if(i+1>=0 && i+1<n) { if(res[i+1]>res[i]+1) { res[i+1]=res[i]+1; que.push(i+1); } } vector<int> vec=tmp[arr[i]]; int m=vec.size(); for(int k=0;k<m;k++) { int idx=vec[k]; if(res[idx]>res[i]+1) { res[idx]=res[i]+1; que.push(idx); } } vector<int>().swap(tmp[arr[i]]); } return res[n-1]; } };