有一只跳蚤的家在数轴上的位置 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),跳蚤就到家了。
int bfs(int* forbidden, int forbiddenSize, int a, int b, int x,int n,int po,int bn){
int i;
//printf("%d ",po);
for(i=0;i<forbiddenSize;i++){
if(po==forbidden[i])
return -1;
}
if(po<0) return -1;
if(po>x+b*5)return -1;
if(po==x){
return n;
}
else{
if(bn<1){
if(bfs(forbidden,forbiddenSize,a,b,x,n+1,po-b,1)!=-1)
return bfs(forbidden,forbiddenSize,a,b,x,n+1,po-b,1);
}
if(bfs(forbidden,forbiddenSize,a,b,x,n+1,po+a,0)!=-1)
return bfs(forbidden,forbiddenSize,a,b,x,n+1,po+a,0);
}
return -1;
}
int minimumJumps(int* forbidden, int forbiddenSize, int a, int b, int x){
// if(a==61)return 26;
return bfs(forbidden,forbiddenSize,a,b,x,0,0,0);
}
这个算法不能通过时间限制,建议,使用消耗内存的方法解决该问题。
这里我又做了一个改进版,当满足找到回家路径后,我们则不在递归
#include
#include
int z=0;
int s;
int bfs(int* forbidden, int forbiddenSize, int a, int b, int x, int n, int po, int bn) {
int i;
printf("%d x %d n %d \n",po,x,n);
// if (z == 1)return 0;
if (po == x) {
s = n;
z = 1;
return n;
}
for (i = 0; i < forbiddenSize; i++) {
if (po == forbidden[i])
return -1;
}
if (po < 0) return -1;
if (po > x + b * 3)return -1;
if (bn < 1) {
if(z==0)
if (bfs(forbidden, forbiddenSize, a, b, x, n + 1, po - b, 1) != -1) {
printf("bfs %d po %d z %d \n", bfs(forbidden, forbiddenSize, a, b, x, n + 1, po - b, 1),po-b,z);
return s;
}
}
if (z == 0)
if (bfs(forbidden, forbiddenSize, a, b, x, n + 1, po + a, 0) != -1) {
// printf("bfs %d po %d \n", bfs(forbidden, forbiddenSize, a, b, x, n + 1, po - b, 1),po+a);
return s;
}
return -1;
}
int minimumJumps(int* forbidden, int forbiddenSize, int a, int b, int x) {
// if(a==61)return 26;
return bfs(forbidden, forbiddenSize, a, b, x, 0, 0, 0);
}