HDU 5025 Saving Tang Monk(状态转移， 广搜)

#include<bits/stdc++.h>
using namespace std;
const int maxN = ;
const int inf = 1e9 + ;
char G[maxN][maxN], snake[maxN][maxN];
int times[maxN][maxN][];
int n, m, sx, sy, ex, ey, ans;
int dir[][] = {{,},{,},{,-},{-,}};
 
struct node
{
    int x, y, t, key, killed; //坐标， 步数， 已经拿到的钥匙， 有没有杀蛇
    bool operator < (const node& p) const
    {
        return t > p.t;
    }
    node(int x, int y, int t, int key, int killed):x(x),y(y), t(t), key(key), killed(killed) {};
};
void bfs()
{
    node t = (node)
    {
        sx, sy, , ,
    };
    times[sx][sy][] = ;
    queue<node> q;
    q.push(t);
    while(!q.empty())
    {
        node u = q.front();
        q.pop();
        int x = u.x, y = u.y, key = u.key, t = u.t, killed = u.killed;
        if(x == ex && y == ey && key == m)
        {
            ans = min(ans, t);
            continue;
        }
        for(int d = ; d < ; d++)
        {
            int tx = x + dir[d][];
            int ty = y + dir[d][];
            if(tx <  || tx >= n || ty <  || ty >= n || G[tx][ty] == '#') //越界
                continue;
 
            if(G[tx][ty] >= '' && G[tx][ty] <= '')   //如果是钥匙
            {
                int num = G[tx][ty] - '';
                if(key +  == num && (times[tx][ty][key + ] > t + ))   //可以拿
                {
                    q.push(node(tx,ty,t+,key+,killed));
                    times[tx][ty][key + ] = t + ;
                }
                else if(((key + ) != num) && (times[tx][ty][key] > t + )) //不可以拿
                {
                    q.push(node(tx,ty,t+,key,killed));
                    times[tx][ty][key] = t + ;
                }
            }
            else if(G[tx][ty] >= 'A' && G[tx][ty] <= 'E') //蛇
            {
                int cnt = G[tx][ty] - 'A'; //压位判断
 
                if((killed & (<<cnt)) == )   //未杀
                {
                    if(times[tx][ty][key] > t + )
                    {
                        q.push(node(tx,ty,t+,key,(killed | (<<cnt))));
                        times[tx][ty][key] = t + ;
                    }
                }
                else     //已经杀了
                {
                    if(times[tx][ty][key] > t + )
                    {
                        q.push(node(tx,ty,t+,key,killed));
                        times[tx][ty][key] = t + ;
                    }
                }
            }
            else if((G[tx][ty] == '.' || G[tx][ty] == 'T' || G[tx][ty] == 'K')  && times[tx][ty][key] > t + )
            {
                q.push(node(tx,ty,t+,key,killed));
                times[tx][ty][key] = t + ;
            }
        }
    }
}
int main()
{
//    freopen("1.txt","r", stdin);
    while(~scanf("%d %d", &n, &m))
    {
        int cnt = ;
        if(n == )
            break;
        for(int i = ; i < n; i++)
            for(int j = ; j < n; j++)
                for(int k = ; k < ; k++)
                    times[i][j][k] = inf;
        for(int i = ; i < n; i++)
        {
            scanf("%s", G[i]);
        }
        for(int i = ; i < n; i++)
        {
            for(int j = ; j < n; j++)
            {
                if(G[i][j] == 'K')
                    sx = i, sy = j;
                if(G[i][j] == 'T')
                    ex = i, ey = j;
                if(G[i][j] == 'S')
                {
                    G[i][j] = (cnt++ + 'A');
                }
            }
        }
 
        ans = inf;
        bfs();
        if(ans == inf)
        {
            printf("impossible\n");
        }
        else
        {
            printf("%d\n", ans);
        }
    }
    return ;
}