最优性剪枝

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 楼至少要按几次按钮呢？

输入格式

共二行。

第一行为三个用空格隔开的正整数，表示 N,A,B（1≤N≤200，1≤A,B≤N）。

第二行为 N 个用空格隔开的非负整数，表示 Ki​。

输出格式

一行，即最少按键次数，若无法到达，则输出 -1。

5 1 5
3 3 1 2 5

3

这道题其实比较简单，根据每一层的数字进行上下搜索就行了，但要注意边界。

#include<iostream>
using namespace std;
int n,a,b;
int ans=1e9;
int d[10000];
void dfs(int x,int sum)
{
	if(sum>=ans)
	{
		return;
	}
	if(x==b)
	{
		ans=min(ans,sum);
		return;
	}
	if(x+d[x]<=n)
	{
		dfs(x+d[x],sum+1);//上
	}
	if(x-d[x]>0)
	{
		dfs(x-d[x],sum+1);//下
	}
}
int main()
{
	cin>>n>>a>>b;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&d[i]);
	}
	dfs(a,0);
	cout<<ans;
}

但是按照普通方式写出来的代码只能过3个点，这就需要用到今天的主题——最优性剪枝。

优化

最优性剪枝是基于方法的最优性进行剪枝，请看以下数据：

要是数据为1 2 4 2 3 5，方法1,2都可以到达，但是方法2明显优于方法1，因为它每一个楼层都只走了一次，为最优方案，因此，我们可以加一个bool类型的数组，只允许访问每个楼层一次，就可以得到最优方案。

#include<iostream>
using namespace std;
int n,a,b;
int ans=1e9;
int d[10000];
bool p[10000];
void dfs(int x,int sum)
{
	if(sum>=ans)
	{
		return;
	}
	if(x<0||x>n)
	{
		return;
	}
	if(x==b)
	{
		ans=min(ans,sum);
		return;
	}
	if(x+d[x]<=n&&!p[x+d[x]])
	{
		p[x+d[x]]=true;
		dfs(x+d[x],sum+1);
		p[x+d[x]]=false;
	}
	if(x-d[x]>0&&!p[x-d[x]])
	{
		p[x-d[x]]=true;
		dfs(x-d[x],sum+1);
		p[x-d[x]]=false;
	}
}
int main()
{
	cin>>n>>a>>b;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&d[i]);
	}
	dfs(a,0);
	if(ans==1e9)
	{
		printf("-1\n");
	}
	else
	printf("%d",ans);
}

这样已经可以过掉大部分的点。

说明

本文引用此题用dfs只是为了介绍一种新的思考方式，但这题数据量确实挺大，不好dfs，所以还是推荐用bfs。

这里给出bfs核心代码：

void bfs(){
	queue<node> q;
	q.push((node){a,0});
	while(!q.empty()){
		if(q.front().num==b){
			cout<<q.front().j;
			return ;
		}
		int x1=q.front().num+arr[q.front().num];
		int x2=q.front().num-arr[q.front().num];
		if(x1>=1 && x1<=n && !f[x1]){
			f[x1]=true;
			q.push((node){x1,q.front().j+1});
		}
		if(x2>=1 && x2<=n && !f[x2]){
			f[x2]=true;
			q.push((node){x2,q.front().j+1});
		}
		q.pop();
	}
	cout<<"-1";
	return ;
}

如果要使用这串代码，别忘了先定义一个结构体node：

struct node{
	int num;
	int j;
}