首页 > 代码库 > HDU 5025

HDU 5025

http://acm.hdu.edu.cn/showproblem.php?pid=5025

蛇最多只有五条,状态压缩一下,vis增加两维,表示取得钥匙的状态和蛇的状态,然后一个优先队列的bfs即可解决问题,简单题

#include <iostream>#include <cstdio>#include <cstring>#include <map>#include <queue>#include <algorithm>using namespace std ;int n,m ;int sx,sy ;int tx,ty ;char mp[105][105] ;int vis[105][105][10][33] ;struct node{    int x,y ;    int time ;    int snake ;    int key ;    friend bool operator <(node a,node b)    {        return a.time>b.time ;    }} ;int cnt ;struct sk{    int x,y ;}ss[10] ;int OK(node a){    if(a.x==tx && a.y==ty)    {        if(a.key==m)return 1 ;    }    return 0 ;}int dx[]={1,-1,0,0} ;int dy[]={0,0,1,-1} ;int bfs(int x,int y){    priority_queue <node> q ;    node s ;    s.x=x ;s.y=y ;s.time=0 ;s.snake=0 ;s.key=0 ;    vis[x][y][0][0]=1 ;    q.push(s) ;    while(!q.empty())    {        node u=q.top() ;        if(OK(u))return u.time ;        q.pop() ;        for(int i=0 ;i<4 ;i++)        {            int xx=u.x+dx[i] ;            int yy=u.y+dy[i] ;            if(xx<0 || xx>=n || yy<0 || yy>=n)continue ;            if(mp[xx][yy]==#)continue ;            if(vis[xx][yy][u.key][u.snake])continue ;            node T ;            T=u ;            T.x=xx ;T.y=yy ;T.time=u.time+1 ;            if(mp[xx][yy]>=1 && mp[xx][yy]<=9)//遇到钥匙            {                int num=mp[xx][yy]-0-1 ;                if(T.key==num)                {                    T.key++ ;                    vis[xx][yy][T.key][T.snake]=1 ;                    q.push(T) ;                }                else                {                    vis[xx][yy][T.key][T.snake]=1 ;                    q.push(T) ;                }            }            else if(mp[xx][yy]==S)//遇到蛇            {                for(int i=0 ;i<cnt ;i++)                {                    if(xx==ss[i].x && yy==ss[i].y)                    {                        if((T.snake>>i)&1)                        {                            vis[xx][yy][T.key][T.snake]=1 ;                            q.push(T) ;                        }                        else                        {                            T.time++ ;                            T.snake+=(1<<i) ;                            vis[xx][yy][T.key][T.snake]=1 ;                            q.push(T) ;                        }                        break ;                    }                }            }            else            {                vis[xx][yy][T.key][T.snake]=1 ;                q.push(T) ;            }        }    }    return -1 ;}int main(){    while(~scanf("%d%d",&n,&m))    {        if(!n && !m)break ;        for(int i=0 ;i<n ;i++)            scanf("%s",mp[i]) ;        cnt=0 ;        for(int i=0 ;i<n ;i++)        {            for(int j=0 ;j<n ;j++)            {                if(mp[i][j]==K)sx=i,sy=j ;                if(mp[i][j]==T)tx=i,ty=j ;                if(mp[i][j]==S)                {                    ss[cnt].x=i ;                    ss[cnt++].y=j ;                }            }        }        memset(vis,0,sizeof(vis)) ;        int ans=bfs(sx,sy) ;        if(ans==-1)puts("impossible") ;        else printf("%d\n",ans) ;    }    return 0 ;}
View Code

 

HDU 5025