小记:忘记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; }