uva1601(雙向BFS經典題)
阿新 • • 發佈:2019-02-14
單向:
/*
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;
}
最後一定要注意,每次擴充套件是擴充套件一層而不是一個點。網上有些程式碼寫法是擴充套件一個點,實際上是錯誤的,應該擴充套件一層。!!擴充套件一個點能過的題目可能是測試資料設計不夠刁。一定要注意!!!!!!