题目描述
呵呵,有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第ii层楼(1≤i≤N)上有一个数字Ki(0≤Ki≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3,3,1,2,5代表了Ki(K1=3,K2=3,…),从11楼开始。在1楼,按“上”可以到4楼,按“下”是不起作用的,因为没有−2楼。那么,从A楼到B楼至少要按几次按钮呢?
输入格式
共二行。
第一行为33个用空格隔开的正整数,表示N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为N个用空格隔开的非负整数,表示Ki。
输出格式
一行,即最少按键次数,若无法到达,则输出−1。
输入输出样例
输入 #1
5 1 5
3 3 1 2 5
输出 #1
3
我的代码:
DFS:
(可以看成二叉树遍历就很好理解了)
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 0xffffff;
int n,a,b,num[210],mincnt = maxn,flag[210] = {false};
void DFS(int index,int cnt)
{
if (index == b) mincnt = min(mincnt,cnt);
if (cnt >= mincnt) return ;
flag[index] = true; //标记表示遍历过,如果再遍历则会重复
if (index - num[index] >= 1 && !flag[index - num[index]]) DFS(index - num[index],cnt+1);
if (index + num[index] <= n && !flag[index + num[index]]) DFS(index + num[index],cnt+1);
flag[index] = false;//回溯,状态还原,其他路径可能会经过
}
int main()
{
scanf("%d%d%d",&n,&a,&b);
for (int i = 1 ;i <= n ;i++)
scanf("%d",&num[i]);
DFS(a,0);
if (mincnt != maxn) printf("%d",mincnt);
else printf("-1");
return 0;
}
BFS:
(BFS是一层一层遍历的,所以如果第一次遇到目的地,那一定是最少次数,时间明显比DFS快了许多)
#include<cstdio>
#include<queue>
using namespace std;
const int maxn = 210;
struct Node
{
int floor,cnt = 0;
}node;
int n,a,b,num[maxn],mincnt = -2;
bool flag[maxn] = {false};
void BFS()
{
queue<Node> q;
node.floor = a;
flag[a] = true;
q.push(node);
while (!q.empty())
{
Node temp = q.front(),p = temp;
q.pop();
if (temp.floor == b)
{
mincnt = temp.cnt;
break;
}
p.floor += num[temp.floor],p.cnt = temp.cnt + 1;
if (!flag[p.floor] && p.floor <= n)
{
flag[p.floor] = true;
q.push(p);
}
p.floor -= 2 * num[temp.floor];
if (!flag[p.floor] && p.floor >= 1)
{
flag[p.floor] = true;
q.push(p);
}
}
}
int main()
{
scanf("%d%d%d",&n,&a,&b);
for (int i = 1 ;i <= n ;i++)
scanf("%d",&num[i]);
BFS();
if (mincnt != -2) printf("%d",mincnt);
else printf("-1");
return 0;
}