P1135 奇怪的电梯 （DFS&BFS双解）（水题）

链接: 原题链接

输入输出样例

输入 #1

5 1 5
3 3 1 2 5

输出 #1

3

思路讲解

本题有很多做法，BFS、DFS、floyd、spfa等等，下面以BFS和DFS为例。
思路很简单，从起点开始，只要没越界就向上下搜，全部搜完得到答案。
注意：DFS需要剪枝，不然会TLE或MLE。

代码实现

BFS：

#include<bits/stdc++.h>
using namespace std;
const int MAXN=205;
int n,st,ed;
int k[MAXN];
int v[MAXN];
int minn=99999999;
int flag;
int d[2]={1,-1};
void BFS(int floor,int ct);
struct node{
	int f,ct;
} start, temp;;
queue<node> r;
int main()
{
	scanf("%d%d%d",&n,&st,&ed);
	for(int i=1;i<=n;i++)
		scanf("%d",&k[i]);
	BFS(st,0);
	if(flag) printf("%d",minn);
	else printf("-1");
}
void BFS(int floor,int ct)
{
	v[floor]=1;
	start.f=floor;
	start.ct=ct;
	r.push(start);
	while(!r.empty())
	{
		int rf=r.front().f;
//		if(r.front().ct>minn)continue;		
		if(rf==ed)
		{
			flag=1;
			if(minn>r.front().ct)
				minn=r.front().ct;
			break;
		}
		for(int i=0;i<=1;i++)
		{
			temp.f=rf+k[rf]*d[i];
			temp.ct=r.front().ct+1;
			if(temp.f>n||temp.f<1) continue;
			if(v[temp.f]==0)
			{
				r.push(temp);
				v[temp.f]=1;
			}
		}
		r.pop();
	}
}

DFS：

#include<bits/stdc++.h>
using namespace std;
const int MAXN=205;
int n,st,ed;
int k[MAXN],v[MAXN];
int ans=99999,flag;
int d[2]={1,-1};//上下方向 
void DFS(int floor,int ct); //在floor层，已经按了ct次 
int main()
{
	scanf("%d%d%d",&n,&st,&ed);
	for(int i=1;i<=n;i++)
		scanf("%d",&k[i]); 
	flag=0;
	v[st]=1;
	DFS(st,0);
	if(flag) printf("%d",ans);
	else printf("-1");
}
void DFS(int floor,int ct) //在floor层，已经按了ct次 
{
	if(floor==ed)
	{
		flag=1;
		if(ct<ans)
			ans=ct;
		return;
	}
	if(ct>ans)return;//剪枝
	int tf;
	v[floor]=1;
	for(int i=0;i<=1;i++)
	{
		tf=floor+k[floor]*d[i]; 
		if(tf>n||tf<1)continue;
		if(v[tf]==0)
		{
			DFS(tf,ct+1);
		}
	}
	v[floor]=0;	 
}