小記:忘記memset wa了一次。。。
思路:每次有兩個方向,上或下。 bfs之,最初碰到的點都是次數最少。這是bfs的特性-最短路。
加個vis標記是否在隊列以及是否已經處理過了。並且越界判斷 和入隊判斷
代碼:
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <map>
#include <set>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
using namespace std;
#define mst(a,b) memset(a,b,sizeof(a))
#define REP(a,b,c) for(int a = b; a < c; ++a)
#define eps 10e-8
const int MAX_ = 10010;
const int N = 100010;
const int INF = 0x7fffffff;
int a, b, n;
bool vis[MAX_];
int h[MAX_];
int to[MAX_];
int dir[2] = {1,-1};
int bfs(int start)
{
queue<int >q;
q.push(start);
q.push(0);
vis[start] = 1;
while(!q.empty()){
int cur = q.front();q.pop();
int cnt = q.front(); q.pop();
if(cur == b)return cnt;
REP(i, 0, 2){
int next = dir[i]*to[cur] + cur;
int tmp = cnt+1;
if(next > 0 && next <= n){
if(!vis[next]){
vis[next] = 1;
q.push(next);
q.push(tmp);
}
}
}
}
return -1;
}
int main()
{
int T, m;
while(scanf("%d", & n), n){
scanf("%d%d", &a, &b);
REP(i, 1, n+1){
scanf("%d", &to[i]);
}
mst(vis, 0);
int tmp = bfs(a);
printf("%d\n", tmp);
}
return 0;
}