DFS-减枝的妙处

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第ii层楼(1≤i≤N)上有一个数字K i (0≤Ki ≤N)。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如：3, 3 ,1 ,2 ,53,3,1,2,5代表了K_i(K_1=3,K_2=3,…)K i(K 1​=3,K 2=3,…)，从11楼开始。在11楼，按“上”可以到44楼，按“下”是不起作用的，因为没有-2−2楼。那么，从AA楼到BB楼至少要按几次按钮呢？

输入格式

共二行。
第一行为33个用空格隔开的正整数，表示N,A,B(1≤N≤200, 1≤A,B≤N)N,A,B(1≤N≤200,1≤A,B≤N)。
第二行为NN个用空格隔开的非负整数，表示K_iK
i 。

输出格式

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

输入输出样例

输入:

5 1 5
3 3 1 2 5

输出:

3

说明：本题利用DFS算法进行求解，但注意有方向，即向上或向下，这里我用dir数组来控制，还有一个最重要的问题，就是如何减去枝叶，把没必要的查询去掉，这里只要后面按压的次数比最小的次数大就退出即可。
代码如下：

#include<iostream>
#include<cstring>
using namespace std;
int K[1001];
int visited[1001];
int dir[2]={-1,1};
int N,A,B;
int press_num=999999;
void dfs(int floor,int count)
{
	int i,j;
	if(press_num<count)//减枝
		return;
	if(floor==B)
	{
		if(press_num>count)
			press_num=count;
		return;
	}
		for(j=0;j<2;j++)
		{
			if(floor+(dir[j]*K[floor])>0&&!visited[floor])
			{
				visited[floor]=1;
				dfs(floor+(dir[j]*K[floor]),count+1);
				visited[floor]=0;
			}
		}
}
int main()
{
	int i;
	cin >> N >> A >> B;
	for(i=1;i<=N;i++)
		cin >> K[i];
	memset(visited,0,sizeof(visited));
	//visited[A]=1;
	dfs(A,0);
	if(press_num!=999999) 
		cout << press_num << endl;
	else
		cout << "-1" << endl;
	return 0;
}