hdu1242 Rescue 优先队列+BFS

#include
#include
#include
using namespace std;


struct node
{
    int x,y,step;
    friend bool operator<(node n1,node n2)//友元函数 优先队列需要用到 
    {
       return n1.step>n2.step;
    }
};


int n,m,vis[205][205];
char map[205][205];
int x1,x2,y1,y2;
int to[4][2] = {1,0,-1,0,0,1,0,-1};


int check(int x,int y)//判断当前状态合法性 
{
    if(x<0 || y<0 || x>=n || y>=m || !vis[x][y] || map[x][y] == '#')
    return 1;
    return 0;
}


int bfs()//bfs只进行一次,不进行dfs那种递进 
{
    int i,ee=1;
    priority_queue Q;
    
    node a,next;
    a.x=x1;
    a.y=y1;
    a.step=0;
    Q.push(a);
    vis[x1][y1]=0;
    while(!Q.empty())//如果队列为空,则返回true,否则返回false 
    {    
        a=Q.top();//返回优先队列对顶元素
        Q.pop();//删除队首元素,但不返回其值
        if(a.x==x2&&a.y==y2)//到达终点,退出 
        return a.step;
        for(i=0;i<4;i++)
        {   
            next=a;
            next.x+=to[i][0];
            next.y+=to[i][1];
            if(check(next.x,next.y))
            continue;
            next.step++;
            if(map[next.x][next.y]=='x')//如果是守卫,需要多花走一步的时间 
            next.step++;
            if(vis[next.x][next.y]>=next.step)//存入最小时间。剪枝 
            {
                vis[next.x][next.y]=next.step;//用优先队列就是想找时间最少的 
                Q.push(next);//在基于优先级的适当位置插入新元素
            }
        }
    }
    return 0;
}


int main()
{
    int i,j;
    while(~scanf("%d%d",&n,&m))
    {
        for(i=0;i        {
            scanf("%s",map[i]);
            for(j=0;map[i][j];j++)
            {
                if(map[i][j]=='r')
                {
                    x1=i;
                    y1=j;
                }
                else if(map[i][j]=='a')
                {
                    x2=i;
                    y2=j;
                }
            }
        }
        memset(vis,1,sizeof(vis));
        int ans=0;
        
        ans=bfs();
        if(ans)
        printf("%d\n",ans);
        else
        printf("Poor ANGEL has to stay in the prison all his life.\n");
    }
    return 0;
}

你可能感兴趣的:(搜索)