洛谷P1135题解 - 技术栈

题目背景

感谢 @yummy 提供的一些数据。

题目描述

呵呵，有一天我做了一个梦，梦见了一种很奇怪的电梯。大楼的每一层楼都可以停电梯，而且第 i 层楼（1≤i≤N）上有一个数字 K[i]（0≤K[i]≤N）。电梯只有四个按钮：开，关，上，下。上下的层数等于当前楼层上的那个数字。当然，如果不能满足要求，相应的按钮就会失灵。例如： 3,3,1,2,5 代表了 K[i]（K[1]=3，K[2]=3，......），从 1 楼开始。在 1 楼，按"上"可以到 4 楼，按"下"是不起作用的，因为没有 −2 楼。那么，从 A 楼到 B 楼至少要按几次按钮呢？

输入格式

共二行。

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

第二行为 N 个用空格隔开的非负整数，表示 K[i]。

输出格式

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

输入输出样例

输入 #1

复制代码

5 1 5
3 3 1 2 5

输出 #1

复制代码

说明/提示

对于 100% 的数据，1≤N≤200，1≤A,B≤N，0≤K[i]≤N。

本题共 16 个测试点，前 15 个每个测试点 6 分，最后一个测试点 10 分。

思路

从 A 开始，搜索每一条路，可以更新答案再继续搜索，不然会搜索多余导致 TLE(超时)。

AC Code

cpp 复制代码

#include<bits/stdc++.h>
using namespace std;
int n,a,b,k[201],dis[201];
void dfs(int node,int step){
	dis[node]=step;
	int v=node-k[node];
	if(1<=v&&step+1<dis[v])
		dfs(v,step+1);
	v=node+k[node];
	if(v<=n&&step+1<dis[v])
		dfs(v,step+1);
	return;
}
int main(){
	memset(dis,0x3f,sizeof(dis));
	cin>>n>>a>>b;
	for(int i=1;i<=n;i++)
		cin>>k[i];
	dfs(a,0);
	cout<<(dis[b]==0x3f3f3f3f?-1:dis[b]);
	return 0;
}