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