搜尋進階-----I
阿新 • • 發佈:2018-12-12
可憐的公主在一次次被魔王擄走一次次被騎士們救回來之後,而今,不幸的她再一次面臨生命的考驗。魔王已經發出訊息說將在T時刻吃掉公主,因為他聽信謠言說吃公主的肉也能長生不老。年邁的國王正是心急如焚,告招天下勇士來拯救公主。不過公主早已習以為常,她深信智勇的騎士LJ肯定能將她救出。 現據密探所報,公主被關在一個兩層的迷宮裡,迷宮的入口是S(0,0,0),公主的位置用P表示,時空傳輸機用#表示,牆用表示,平地用.表示。騎士們一進入時空傳輸機就會被轉到另一層的相對位置,但如果被轉到的位置是牆的話,那騎士們就會被撞死。騎士們在一層中只能前後左右移動,每移動一格花1時刻。層間的移動只能通過時空傳輸機,且不需要任何時間。 Input 輸入的第一行C表示共有C個測試資料,每個測試資料的前一行有三個整數N,M,T。 N,M迷宮的大小N
….P #.… **… …. *.#… Sample Output YES
由於wa和tle了5發,所以特來記錄一下,簡單搜尋,但是有坑 坑點: 1>只要在小於等於t內救出公主即可 2>如果兩層都是#,這樣dfs會超時,就是沒考慮到這所以超時了 要做剪枝找的深度大於t就不用再搜了
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<map> #include<queue> #include<cmath> using namespace std; typedef long long LL; #define pb push_back #define mp make_pair #define fi first #define se second const int inf = 0x3f3f3f3f; char s[2][15][15]; bool vis[2][15][15]; int col[5] = {0,0,-1,1}; int con[5] = {-1,1,0,0}; int n,m,k; int sx,sy,sz; bool flag = false; bool judge(int i,int j) { if(i < 0 || j < 0) return false; if(i >= n || j >= m) return false; return true; } void dfs(int x,int y,int z,int len) { //cout << x << " " << y << " " << z << " " << len << endl; if(flag) return ; if(len > k) return ; if(x == sx && y == sy && z == sz){ //cout << 0 << endl; if(len <= k){ flag = true; } return; } for(int i = 0;i < 4;++i) { int y1 = y + col[i]; int z1 = z + con[i]; if(judge(y1,z1) && !vis[x][y1][z1]){ if(s[x][y1][z1] == '#'){ vis[x][y1][z1] = true; vis[1 - x][y1][z1] = true; dfs(1 - x,y1,z1,len + 1); vis[x][y1][z1] = false; vis[1 - x][y1][z1] = false; } if(s[x][y1][z1] == '.' || s[x][y1][z1] == 'P'){ vis[x][y1][z1] = true; dfs(x,y1,z1,len + 1); vis[x][y1][z1] = false; } } } } int main() { int t; scanf("%d",&t); while(t--) { scanf("%d %d %d",&n,&m,&k); for(int i = 0;i < 2;++i) { for(int j = 0;j < n;++j){ scanf("%s",s[i][j]); //printf("%s\n",s[i][j]); } } for(int i = 0;i < n;++i) { for(int j = 0;j < m;++j) { if(s[0][i][j] == '#' && s[1][i][j] == '*'){ s[0][i][j] = '*'; } if(s[0][i][j] == '*' && s[1][i][j] == '#'){ s[1][i][j] = '*'; } if(s[0][i][j] == '#' && s[1][i][j] == '#'){ s[1][i][j] = '*'; s[0][i][j] = '*'; } if(s[0][i][j] == 'P'){ sx = 0;sy = i;sz = j; } if(s[1][i][j] == 'P'){ sx = 1;sy = i;sz = j; } } } memset(vis,false,sizeof(vis)); //cout << sx << " " << sy << " " << sz << endl; flag = false; dfs(0,0,0,0); if(flag){ printf("YES\n"); }else{ printf("NO\n"); } } return 0; }