问题 E: 奇怪的电梯
时间限制: 1 Sec 内存限制: 64 MB
提交: 35 解决: 16
[提交][状态][讨论版]
题目描述
有一天桐桐做了一个梦,梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯,而且第i层楼(1≤i≤N)上有一个数字K;(0≤Ki≤N)。电梯只有四
个按钮:开,关,上,下。上下的层数等于当前楼层上的那个数字。当然,如果不能满足要求,相应的按钮就会失灵。例如:3 3 1 2 5代表了Ki
(K1=3,K2=3,…),从一楼开始。在一楼,按“上,”可以到4楼,按“下”是不起作用的,因为没有-2楼。那么,从A楼到B楼至少要按几次按钮
呢?
输入
第1行为三个正整数,表示N,A,B(1≤N≤200,1≤A,B≤N);
第2行为N个正整数,表示Ki。
输出
1行,即最少按键次数,若无法到达,则输出-1。
样例输入
5 1 5
3 3 1 2 5
样例输出
3
先试试Dijkstra,发现过了,代码如下
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std; const int inf = <<; int n;
int map[][];
int a[],cnt;
int vis[],cast[]; void Dijkstra(int s,int e)
{
int i,j,min,pos;
memset(vis,,sizeof(vis));
for(i = ; i<n; i++)
cast[i] = map[s][i];
cast[s] = ;
vis[s] = ;
for(i = ; i<n; i++)
{
min = inf;
for(j = ; j<n; j++)
{
if(cast[j]<min && !vis[j])
{
pos = j;
min = cast[j];
}
}
if(min == inf)
break;
vis[pos] = ;
for(j = ; j<n; j++)
{
if(cast[pos]+map[pos][j]<cast[j] && !vis[j])
cast[j] = cast[pos]+map[pos][j];
}
}
} int main()
{
int i,j,s,e,x,y;
scanf("%d",&n);
scanf("%d%d",&s,&e);
s--,e--;
for(i = ; i<n; i++)
for(j = ; j<n; j++)
map[i][j] = inf;
for(i = ; i<n; i++)
{
scanf("%d",&a[i]);
if(i+a[i]<n)
map[i][i+a[i]] = ;
if(i-a[i]>=)
map[i][i-a[i]] = ;
}
Dijkstra(s,e);
printf("%d\n",cast[e]==inf?-:cast[e]);
return ;
}
后来用BFS交了一发,也过了。
#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<map>
#include <queue>
#define ll long long
#define rep(i,m,n) for(i=m;i<=n;i++)
#define inf 0x3f3f3f3f
const int maxn=1e3+;
using namespace std;
int n,m,a,b;
const int dis[][]={,,,,-,,,-};
int q[maxn],vis[maxn];
int main()
{
int i,j,k,t;
queue<int>p;
scanf("%d%d%d",&n,&a,&b);
vis[a]=;
for(i=;i<=n;i++)scanf("%d",&q[i]);
p.push(a);
while(!p.empty())
{
int now=p.front(),l=now-q[now],r=now+q[now];
p.pop();
if(l>=&&!vis[l])p.push(l),vis[l]=vis[now]+;
if(r<=n&&!vis[r])p.push(r),vis[r]=vis[now]+;
if(l==b||r==b)break;
}
printf("%d\n",vis[b]-);
//system("pause");
return ;
}