来源:https://www.luogu.org/problemnew/show/P1135
题目描述
有一天我做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第i层楼(1≤i≤N)上有一个数字Ki(0≤Ki≤N)。电梯只有四个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3, 3 ,1 ,2 ,5代表了Ki(K1=3,K2=3,…),从1楼开始。在1楼,按“上”可以到4楼,按“下”是不起作用的,因为没有−2楼。那么,从A楼到B楼至少要按几次按钮呢?
输入
共二行。
第一行为3个用空格隔开的正整数,表示N,A,B(1≤N≤200, 1≤A,B≤N) ,第二行为N个用空格隔开的非负整数,表示K i。
输出
一行,即最少按键次数,若无法到达,则输出−1。
样例输入
5 1 5
3 3 1 2 5
样例输出
3
解法一:BFS
AC_code:
#include <bits/stdc++.h>
using namespace std;
int n,a,b;
int g[205];
bool vis[205];
int f[205],flag;
int BFS()
{
queue<int>q;
q.push(a);
f[a] = 0;
while(!q.empty())
{
int k = q.front();
if(k == b)
{
return f[k];
}
//vis[k] = true;
//vis单独放在这标记是不对的,Ac90%,应该是入队就标记
q.pop();
int x = k + g[k], y = k - g[k];
if(x<=n && !vis[x])
{
vis[x] = true;
q.push(x);
f[x] = f[k]+1;
}
if(y>=1 && !vis[y])
{
vis[y] = true;
q.push(y);
f[y] = f[k]+1;
}
}
return -1;
}
int main()
{
cin>>n>>a>>b;
memset(f,0,sizeof(f));
memset(vis,false,sizeof(vis));
for(int i = 1; i <= n; i++)
{
cin>>g[i];
}
int ans = BFS();
cout<<ans<<endl;
return 0;
}
解法二:
DFS:
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
vector<int>v[205];
bool vis[205];
int n,a,b,ans = INF ;
void DFS(int s,int cnt)
{
if(cnt >= ans)
return;
if(s == b)
{
ans = cnt;
return;
}
for(int i = 0; i < v[s].size(); i++)
{
if(!vis[v[s][i]])
{
vis[v[s][i]] = true;
DFS(v[s][i],cnt+1);
vis[v[s][i]] = false;
}
}
}
int main()
{
cin>>n>>a>>b;
int x;
for(int i = 1; i <= n; i++)
{
cin>>x;
int y = i + x, z = i - x;
if(y<=n)
v[i].push_back(y);
if(z>=1)
v[i].push_back(z);
}
DFS(a,0);
if(ans == INF)
cout<<-1<<endl;
else
cout<<ans<<endl;
return 0;
}