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[][] ;
int vis[][][][] ;
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[] ;
int OK(node a)
{
    if(a.x==tx && a.y==ty)
    {
        if(a.key==m)return  ;
    }
    return  ;
}
int dx[]={,-,,} ;
int dy[]={,,,-} ;
int bfs(int x,int y)
{
    priority_queue <node> q ;
    node s ;
    s.x=x ;s.y=y ;s.time= ;s.snake= ;s.key= ;
    vis[x][y][][]= ;
    q.push(s) ;
    while(!q.empty())
    {
        node u=q.top() ;
        if(OK(u))return u.time ;
        q.pop() ;
        for(int i= ;i< ;i++)
        {
            int xx=u.x+dx[i] ;
            int yy=u.y+dy[i] ;
            if(xx< || xx>=n || yy< || 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+ ;
            if(mp[xx][yy]>='' && mp[xx][yy]<='')//遇到钥匙
            {
                int num=mp[xx][yy]-''- ;
                if(T.key==num)
                {
                    T.key++ ;
                    vis[xx][yy][T.key][T.snake]= ;
                    q.push(T) ;
                }
                else
                {
                    vis[xx][yy][T.key][T.snake]= ;
                    q.push(T) ;
                }
            }
            else if(mp[xx][yy]=='S')//遇到蛇
            {
                for(int i= ;i<cnt ;i++)
                {
                    if(xx==ss[i].x && yy==ss[i].y)
                    {
                        if((T.snake>>i)&)
                        {
                            vis[xx][yy][T.key][T.snake]= ;
                            q.push(T) ;
                        }
                        else
                        {
                            T.time++ ;
                            T.snake+=(<<i) ;
                            vis[xx][yy][T.key][T.snake]= ;
                            q.push(T) ;
                        }
                        break ;
                    }
                }
            }
            else
            {
                vis[xx][yy][T.key][T.snake]= ;
                q.push(T) ;
            }
        }
    }
    return - ;
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        if(!n && !m)break ;
        for(int i= ;i<n ;i++)
            scanf("%s",mp[i]) ;
        cnt= ;
        for(int i= ;i<n ;i++)
        {
            for(int j= ;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,,sizeof(vis)) ;
        int ans=bfs(sx,sy) ;
        if(ans==-)puts("impossible") ;
        else printf("%d\n",ans) ;
    }
    return  ;
}