Leetcode 跳跃游戏 IV （BFS+剪枝优化）

原题传送门

1345. 跳跃游戏 IV
给你一个整数数组 arr ，你一开始在数组的第一个元素处（下标为 0）。

每一步，你可以从下标 i 跳到下标 i + 1 、i - 1 或者 j ：

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 处，也就是数组的最后一个元素处。
 

提示：

1 <= arr.length <= 5 * 104
-108 <= arr[i] <= 108

class Solution {
public:
    int minJumps(vector<int>& arr) {
        queue<int> q; // 创建一个队列，用于BFS

        int n = arr.size(); // 获取数组的长度
        vector<bool> vis(n, false); // 创建一个布尔数组，标记每个索引是否被访问过
        vector<int> dis(n, n + 5); // 创建一个距离数组，初始化为一个大数（这里用n+5，假设数组长度不会超过n+5）
        dis[0] = 0; // 起始点的距离设置为0

        unordered_map<int, vector<int>> mp; // 创建一个哈希表，键为数组值，值为具有该值的索引列表
        for (int i = 0; i < n; i++) {
            mp[arr[i]].push_back(i); // 填充哈希表
        }

        q.push(0); // 将起始索引0推入队列
        while (!q.empty()) { // 当队列不为空时，进行BFS
            int now = q.front(); // 获取当前访问的索引
            q.pop(); // 从队列中移除
            if (vis[now])
                continue; // 如果当前索引已经访问过，跳过此次循环
            vis[now] = true; // 标记当前索引为已访问

            // 检查当前索引的左右邻居
            if (now + 1 < n && dis[now + 1] > dis[now] + 1) { // 如果右邻居存在且未访问过
                q.push(now + 1); // 将右邻居推入队列
                dis[now + 1] = dis[now] + 1; // 更新右邻居的距离
            }
            if (now - 1 >= 0 && dis[now - 1] > dis[now] + 1) { // 如果左邻居存在且未访问过
                q.push(now - 1); // 将左邻居推入队列
                dis[now - 1] = dis[now] + 1; // 更新左邻居的距离
            }

            // 检查所有值相同的索引
            for (int& j : mp[arr[now]]) {
                if (now != j && dis[j] > dis[now] + 1) { // 如果索引不同且未访问过
                    q.push(j); // 将索引推入队列
                    dis[j] = dis[now] + 1; // 更新索引的距离
                }
            }
            mp[arr[now]].clear(); // 清除已访问的值对应的索引列表，避免后续重复访问
        }
        return dis[n - 1]; // 返回到达最后一个元素的最短距离
    }
};