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