A计划 HDU-2102 搜索

题目

可怜的公主在一次次被魔王掳走一次次被骑士们救回来之后，而今，不幸的她再一次面临生命的考验。魔王已经发出消息说将在T时刻吃掉公主，因为他听信谣言说吃公主的肉也能长生不老。年迈的国王正是心急如焚，告招天下勇士来拯救公主。不过公主早已习以为常，她深信智勇的骑士LJ肯定能将她救出。
现据密探所报，公主被关在一个两层的迷宫里，迷宫的入口是S（0，0，0），公主的位置用P表示，时空传输机用#表示，墙用*表示，平地用.表示。骑士们一进入时空传输机就会被转到另一层的相对位置，但如果被转到的位置是墙的话，那骑士们就会被撞死。骑士们在一层中只能前后左右移动，每移动一格花1时刻。层间的移动只能通过时空传输机，且不需要任何时间。

Input

输入的第一行C表示共有C个测试数据，每个测试数据的前一行有三个整数N，M，T。 N，M迷宫的大小N*M（1 <= N,M <=10)。T如上所意。接下去的前N*M表示迷宫的第一层的布置情况，后N*M表示迷宫第二层的布置情况。

Output

如果骑士们能够在T时刻能找到公主就输出“YES”，否则输出“NO”。

Sample Input

1
5 5 14
S*#*.
.#...
.....
****.
...#.

..*.P
#.*..
***..
...*.
*.#..

Sample Output

YES

分析

典型的搜索，只不过这个是两层而已，因为数很小，所以可以用一个三维数组存地图。有一个坑点就是，当走到‘#’的时候，虽然传到另一层不花费时间，但是走到‘#’也需要一秒的时间，所以进行下一步的搜索时，时间也得加1。再就是优化，把地图中上下两层都是‘#’的，直接全部改成墙，一层是‘#’，另一层是墙的，也把‘#’改成墙，省去了判断被撞死的情况。还需要两处剪枝，如果ans已经小于t了，直接返回；如果step大于t，也直接返回。

代码

#include<iostream>
#include<cstdio>
#include<string.h>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<stdlib.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f
int n,m,t;
int ans;
char mp[2][12][12];
bool vis[2][12][12];
void dfs(int d,int x,int y,int step)
{
	if(ans<=t)//剪枝  
		return ;
    if(step>=ans)//剪枝  
        return ;
    if(mp[d][x][y]=='P')
    {
        ans=min(ans,step);
        return ;
    }
    int nxt[4][2]={0,1,1,0,0,-1,-1,0};
    for(int k=0;k<4;k++)
    {
        int tx=x+nxt[k][0];
        int ty=y+nxt[k][1];
        if(tx<0||tx>=n||ty<0||ty>=m||vis[d][tx][ty])
            continue;
        if(vis[d][tx][ty]==0)
        {
            if(mp[d][tx][ty]=='.'||mp[d][tx][ty]=='P')
            {
                vis[d][tx][ty]=1;
                dfs(d,tx,ty,step+1);
                vis[d][tx][ty]=0;
            }
            if(mp[d][tx][ty]=='#'&&vis[1-d][tx][ty]==0)
            {
                vis[d][tx][ty]=1;
                vis[1-d][tx][ty]=1;
                dfs(1-d,tx,ty,step+1);
                vis[d][tx][ty]=0;
                vis[1-d][tx][ty]=0;
            }
        }
    }
}
int main()
{
    int c,d,sx,sy;
    scanf("%d",&c);
    while(c--)
    {
        scanf("%d%d%d",&n,&m,&t);
        for(int i=0;i<n;i++)
            scanf("%s",&mp[0][i]);
        for(int i=0;i<n;i++)
            scanf("%s",&mp[1][i]);
        for(int i=0;i<2;i++)
        {
            for(int j=0;j<n;j++)
            {
                for(int k=0;k<m;k++)
                {
                    if(mp[i][j][k]=='S')
                    {
                        d=i;
                        sx=j;
                        sy=k;
                    }
                    if(mp[i][j][k]=='#'&&mp[1-i][j][k]=='#')
                    {
                        mp[i][j][k]='*';
                        mp[1-i][j][k]='*';
                    }
                    if(mp[i][j][k]=='#'&&mp[1-i][j][k]=='*')
                    {
                        mp[i][j][k]='*';
                    }
                }
            }
        }
        memset(vis,0,sizeof(vis));
        vis[d][sx][sy]=1;
        ans=inf;
        dfs(d,sx,sy,0);
        if(ans<=t)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}