奇怪的电梯

奇怪的电梯

题目描述

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

输入格式

共二行。

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

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

输出格式

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

样例 #1

样例输入 #1

5 1 5
3 3 1 2 5

样例输出 #1

3

提示

对于 $100\%$ 的数据，$1\le N\le 200$，$1\le A,B\le N$，$0\le K_i\le N$。

---

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Scanner;
import java.io.*;

public class Main {
	
	static int N = 210;
	static int dist[] = new int[N];
	static int k[] = new int[N];
	static int n, a, b;
	
	public static void main(String[] args) throws IOException{
		
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StreamTokenizer stmInput = new StreamTokenizer(br);
		stmInput.nextToken();
		n = (int)stmInput.nval;
		stmInput.nextToken();
		a = (int)stmInput.nval;
		stmInput.nextToken();
		b = (int)stmInput.nval;
		for(int i = 0; i < n; i++) {
			stmInput.nextToken();
			k[i] = (int)stmInput.nval;
		}
		
		bfs();
		System.out.println(dist[b]);
		br.close();
	}
	
	public static void bfs() {
		
		Arrays.fill(dist, 1, n+1, -1); 
		dist[a] = 0;
		LinkedList<Integer> q = new LinkedList<>();
		q.addLast(a);
		
		while(!q.isEmpty()) {
			
			int t = q.removeFirst();
			if(t == b) return;
				
			int l = t - k[t - 1], r = t + k[t - 1];
			if(l > 0 && dist[l] == -1) {
				dist[l] = dist[t] + 1;
				q.addLast(l);
			}
			if(r <= n && dist[r] == -1) {
				dist[r] = dist[t] + 1;
				q.addLast(r);
			}
			
		}
		
	}
	
	
}