1. 程式人生 > >搜尋進階-----I

搜尋進階-----I

可憐的公主在一次次被魔王擄走一次次被騎士們救回來之後,而今,不幸的她再一次面臨生命的考驗。魔王已經發出訊息說將在T時刻吃掉公主,因為他聽信謠言說吃公主的肉也能長生不老。年邁的國王正是心急如焚,告招天下勇士來拯救公主。不過公主早已習以為常,她深信智勇的騎士LJ肯定能將她救出。 現據密探所報,公主被關在一個兩層的迷宮裡,迷宮的入口是S(0,0,0),公主的位置用P表示,時空傳輸機用#表示,牆用表示,平地用.表示。騎士們一進入時空傳輸機就會被轉到另一層的相對位置,但如果被轉到的位置是牆的話,那騎士們就會被撞死。騎士們在一層中只能前後左右移動,每移動一格花1時刻。層間的移動只能通過時空傳輸機,且不需要任何時間。 Input 輸入的第一行C表示共有C個測試資料,每個測試資料的前一行有三個整數N,M,T。 N,M迷宮的大小N

M(1 <= N,M <=10)。T如上所意。接下去的前NM表示迷宮的第一層的佈置情況,後NM表示迷宮第二層的佈置情況。 Output 如果騎士們能夠在T時刻能找到公主就輸出“YES”,否則輸出“NO”。 Sample Input 1 5 5 14 S*#*. .#… … ****. …#.

.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;
}