hdu 1548 A strange lift(水題,bfs)

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


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章