1. 程式人生 > >uva1601(雙向BFS經典題)

uva1601(雙向BFS經典題)

單向:

/*
solution:
    還是隱式圖搜尋問題,類似dijkstra,用bfs解決,這個是單向搜尋,弊端是有時候隨著搜尋
    的深入,可擴充套件的結點會越來越多,造成效率變慢,用雙向bfs可以解決這個問題

note:
    隱式圖的搜尋
    處理輸入時候要用fgets,否則會TLE

date:
    2016/5/5
*/
#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; }

雙向:

#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], color[200][200][200], dist[200][200][200], redist[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> qf;  //記錄正向bfs
    queue<int> qb;  //記錄反向bfs

    dist[s[0]][s[1]][s[2]] = 0;
    dist[t[0]][t[1]][t[2]] = 1; //分別記錄正反兩種遍歷走了多少步數

    qf.push(ID(s[0], s[1], s[2]));
    qb.push(ID(t[0], t[1], t[2]));  //起點終點分別壓入佇列

    color[s[0]][s[1]][s[2]] = 1;
    color[t[0]][t[1]][t[2]] = 2;    //分別標註正反兩種遍歷已經走過的

    while(!qf.empty() || !qb.empty()) {
        int fnum = qf.size(), bnum = qb.size();
        while(fnum--) {
            int u = qf.front(); qf.pop();
            int a = (u >> 16) & 0xff, b = (u >> 8) & 0xff, c = u & 0xff;

            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(color[a2][b2][c2] == 0) {
                            dist[a2][b2][c2] = dist[a][b][c] + 1;
                            color[a2][b2][c2] = 1;
                            qf.push(ID(a2, b2, c2));
                        }
                        else if(color[a2][b2][c2] == 2) {
                            return dist[a][b][c] + dist[a2][b2][c2];
                        }
                    }
                }
            }
        }

        while(bnum--) {
            int u = qb.front(); qb.pop();
            int a = (u >> 16) & 0xff, b = (u >> 8) & 0xff, c = u & 0xff;

            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(color[a2][b2][c2] == 0) {
                            dist[a2][b2][c2] = dist[a][b][c] + 1;
                            color[a2][b2][c2] = 2;
                            qb.push(ID(a2, b2, c2));
                        }
                        else if(color[a2][b2][c2] == 1) {
                            return dist[a][b][c] + dist[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, 0, sizeof(dist));
        memset(color, 0, sizeof(color));

        if(s[0] == t[0] && s[1] == t[1] && s[2] == t[2])    printf("0\n");
        else    printf("%d\n", bfs());
    }
    return 0;
}

最後一定要注意,每次擴充套件是擴充套件一層而不是一個點。網上有些程式碼寫法是擴充套件一個點,實際上是錯誤的,應該擴充套件一層。!!擴充套件一個點能過的題目可能是測試資料設計不夠刁。一定要注意!!!!!!