现在的位置: 首页 > 综合 > 正文

hdu 1548 A strange lift(水题,bfs)

2017年10月18日 ⁄ 综合 ⁄ 共 946字 ⁄ 字号 评论关闭

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

抱歉!评论已关闭.