LeetCode-Python-1345. 跳跃游戏 IV (BFS)

给你一个整数数组 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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/jump-game-iv
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:

找最短步数就上BFS,

初始是在下标为0的点,每一次跳跃可以跳到左侧,右侧或者任意值跟当前值相同的下标处,最终目标是到达最后一个下标。

过程就是标准的BFS,但此题很容易超时,需要做额外处理:

如果有一段连续的区间上所有元素的值相等,那么只需保留左右端点。

比如对于一个输入比如:

[7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 9]来说,如果我们从第一个7开始跳,实际上直接一步跳到最远的7即可,中间的所有7都是无效的。

时间复杂度:O(N)

空间复杂度:O(N)

class Solution(object):
    def minJumps(self, arr):
        """
        :type arr: List[int]
        :rtype: int
        """
        from collections import defaultdict, deque
        dic = defaultdict(list)
        for i, x in enumerate(arr):
            if (i and arr[i] != arr[i - 1]) or (i < len(arr) - 1 and arr[i] != arr[i + 1]): # 剪枝
                dic[x].append(i)

        queue = deque([(0, 0)]) #pos, step 
        visited = set([0])

        while queue:
            pos, step = queue.popleft() 

            if pos == len(arr) - 1:
                return step

            for p in [pos - 1, pos + 1] + dic[arr[pos]]: # 往左跳,往右跳,或者跳相同元素
                if 0 <= p < len(arr) and p not in visited:
                    queue.append((p, step + 1))
                    visited.add(p)
发布了722 篇原创文章 · 获赞 106 · 访问量 19万+

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/104275357