LeetCode 1654. 到家的最少跳跃次数(BFS)

文章目录

1. 题目

有一只跳蚤的家在数轴上的位置 x 处。请你帮助它从位置 0 出发,到达它的家。

跳蚤跳跃的规则如下:

  • 它可以 往前 跳恰好 a 个位置(即往右跳)。
  • 它可以 往后 跳恰好 b 个位置(即往左跳)。
  • 不能 连续跳 2 次。
  • 它不能跳到任何 forbidden 数组中的位置。

跳蚤可以往前跳 超过 它的家的位置,但是它 不能跳到负整数 的位置。

给你一个整数数组 forbidden ,其中 forbidden[i] 是跳蚤不能跳到的位置,同时给你整数 a, b 和 x ,请你返回跳蚤到家的最少跳跃次数。
如果没有恰好到达 x 的可行方案,请你返回 -1 。

示例 1:
输入:forbidden = [14,4,18,1,15], a = 3, b = 15, x = 9
输出:3
解释:往前跳 3 次(0 -> 3 -> 6 -> 9),跳蚤就到家了。

示例 2:
输入:forbidden = [8,3,16,6,12,20], a = 15, b = 13, x = 11
输出:-1

示例 3:
输入:forbidden = [1,6,2,14,5,17,4], a = 16, b = 9, x = 7
输出:2
解释:往前跳一次(0 -> 16),然后往回跳一次(16 -> 7),跳蚤就到家了。
 
提示:
1 <= forbidden.length <= 1000
1 <= a, b, forbidden[i] <= 2000
0 <= x <= 2000
forbidden 中所有位置互不相同。
位置 x 不在 forbidden 中。

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

2. 解题

  • 广度优先搜索,搜索的位置需要比 x 大一些
  • 然后往回跳的时候,注意不用标记已经访问过,往前跳的时候标记访问即可
class Solution {
public:
    int minimumJumps(vector<int>& forbidden, int a, int b, int x) {
        int endpos = x+4000;
    	vector<bool> vis(endpos, false);
    	unordered_set<int> fib(forbidden.begin(), forbidden.end());
    	vis[0] = true;
    	queue<pair<int,bool>> q;
    	q.push({0,false});//位置,上一次是向后的吗
    	int step = 0, size;
    	while(!q.empty()) 
    	{
    		size = q.size();
    		while(size--)
    		{
    			int p = q.front().first;
                bool back = q.front().second;
    			if(p == x)
    				return step;
    			q.pop();
    			if(p+a < endpos && !fib.count(p+a) && !vis[p+a])
    			{
    				vis[p+a] = true;
    				q.push({p+a, false});
    			}
    			if(p-b > 0 && !fib.count(p-b) && !vis[p-b] && !back)
    			{
    				// vis[p-b] = true; //不能写
    				q.push({p-b, true});
    			}
    		}
    		step++;
    	}
    	return -1;
    }
};

56 ms 14.7 MB

或者一个位置,使用两个访问标记,前向的和后向的

class Solution {
public:
    int minimumJumps(vector<int>& forbidden, int a, int b, int x) {
        int endpos = x+4000;
    	vector<vector<bool>> vis(endpos, vector<bool>(2,false));
    	unordered_set<int> fib(forbidden.begin(), forbidden.end());
    	vis[0][0] = vis[0][1] = true;
    	queue<pair<int,bool>> q;
    	q.push({0,false});//位置,上一次是向后的吗
    	int step = 0, size;
    	while(!q.empty()) 
    	{
    		size = q.size();
    		while(size--)
    		{
    			int p = q.front().first;
                bool back = q.front().second;
    			if(p == x)
    				return step;
    			q.pop();
    			if(p+a < endpos && !fib.count(p+a) && !vis[p+a][0])
    			{
    				vis[p+a][0] = true;
    				q.push({p+a, false});
    			}
    			if(p-b > 0 && !fib.count(p-b) && !vis[p-b][1] && !back)
    			{
    				vis[p-b][1] = true; 
    				q.push({p-b, true});
    			}
    		}
    		step++;
    	}
    	return -1;
    }
};

404 ms 49.2 MB


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
Michael阿明

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页