万圣节后的早晨

uva1601 

原文：https://blog.csdn.net/qq_29169749/article/details/51420097 

过两天补一下java的和双向的。。

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>

using namespace std;
int w, h, n, s[3], t[3];
char dataset[20][20];
int G[200][5], vis[200][200][200], dist[200][200][200];
int deg[200];   //记录每个编号为i的空格周围可以走的步数

int dx[] = {0, -1, 1, 0, 0};
int dy[] = {0, 0, 0, -1, 1};

inline int ID(int a, int b, int c) {//记录状态
    return (a << 16) | (b << 8) | c;
}

inline bool conflict(int a, int b, int a2, int b2) {
    return ((a2 == b2) || (a == b2 && b == a2));//同一位置或交换都不行
}

int bfs() {
    queue<int> q;
    q.push(ID(s[0], s[1], s[2]));
    dist[s[0]][s[1]][s[2]] = 0;

    while(!q.empty()) {
        int u = q.front();  q.pop();
        int a = (u >> 16) & 0xff, b = (u >> 8) & 0xff, c = u & 0xff;    //解码出出列状态三个小鬼的位置

        if(a == t[0] && b == t[1] && c == t[2]) return dist[a][b][c];

        for(int i = 0; i < deg[a]; i++) {
            int a2 = G[a][i];
            for(int j = 0; j < deg[b]; j++) {
                int b2 = G[b][j];
                if(conflict(a, b, a2, b2))  continue;
                for(int k = 0; k < deg[c]; k++) {
                    int c2 = G[c][k];
                    if(conflict(a, c, a2, c2) || conflict(b, c, b2, c2))    continue;

                    if(dist[a2][b2][c2] == -1) {    //等于-1说明没有访问过该状态，就要压入队列
                        dist[a2][b2][c2] = dist[a][b][c] + 1;
                        q.push(ID(a2, b2, c2));
                    }

                }
            }
        }
    }
    return -1;
}

int main() {
    //freopen("input.txt", "r", stdin);
    while(~scanf("%d%d%d\n", &w, &h, &n) && n) {
        for(int i = 0; i < h; i++)  fgets(dataset[i], 20, stdin);   //此处不能用scanf来处理输入，会TLE

        int cnt = 0, x[200], y[200], id[20][20];    //从图中抽取出空间并求出初始状态和目标状态
        for(int i = 0; i < h; i++)
        for(int j = 0; j < w; j++) {
            if(dataset[i][j] != '#') {
                x[cnt] = i; y[cnt] = j; id[i][j] = cnt;
                if(islower(dataset[i][j]))  s[dataset[i][j] - 'a'] = cnt;
                else if(isupper(dataset[i][j])) t[dataset[i][j] - 'A'] = cnt;
                cnt++;  //注意这里的cnt++不能偷懒在上面一行末尾，因为这样有时候cnt++会没有执行
            }
        }


        for(int i = 0; i < cnt; i++) {  //利用空格建立图
            deg[i] = 0;
            for(int j = 0; j < 5; j++) {
                int nx = x[i] + dx[j];  int ny = y[i] + dy[j];
                if(dataset[nx][ny] != '#')  G[i][deg[i]++] = id[nx][ny];
            }
        }

        if(n <= 2)  { deg[cnt] = 1; G[cnt][0] = cnt;    s[2] = t[2] = cnt++; }//补充成三个小鬼，并保持补充的小鬼一直在结束位置，
        if(n <= 1)  { deg[cnt] = 1; G[cnt][0] = cnt;    s[1] = t[1] = cnt++; }

        memset(dist, -1, sizeof(dist));

        printf("%d\n", bfs());
    }
    return 0;
}