易保研机试训练营-基础营|搜索|E - A计划

题意

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

Input

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

Output

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

Sample Input

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

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

Sample Output

YES

错误代码

#include 
using namespace std;
int n, m, t;
char map1[10][10];
bool visit1[10][10];
char map2[10][10];
bool visit2[10][10];
int dirx[4] = {0,0,1,-1};
int diry[4] = {1,-1,0,0};
int mint;
void DFS1(int x, int y, int T);
void DFS2(int x, int y, int T) {
	if(visit2[x][y] == 1) return ;
	if(T > t) return ; 
	if(map2[x][y] == 'P') {
		if(T < mint)
			mint = T;
		return ;
	}
	else if(map2[x][y] == '*')
		return ;
	else if(map2[x][y] == '#') {
		if(map1[x][y] != '*' && map1[x][y] != '#') {
			visit2[x][y] = 1;		
			DFS1(x, y, T);
			visit2[x][y] = 0;	
		} 		
	}	
	else if(map2[x][y] == '.') {
		visit2[x][y] = 1;
		for(int i = 0; i < 4; ++i) {
			if(x + dirx[i] < 0 || x + dirx[i] >= n || y + diry[i] < 0 || y + diry[i] >= m)
				continue;
			else {
				if(visit2[x + dirx[i]][y + diry[i]] == 0) {
					DFS2(x + dirx[i], y + diry[i], T + 1);
				}
			}		
		}
		visit2[x][y] = 0;	
	}	
}
void DFS1(int x, int y, int T) {
	if(visit1[x][y] == 1) return ;
	if(T > t)
		return ; 
	if(map1[x][y] == 'P') {
		if(T < mint)
			mint = T;
		return ;
	}	
	else if(map1[x][y] == '#') {
		if(map2[x][y] != '*' && map2[x][y] != '#') {
			visit1[x][y] = 1;		
			DFS2(x, y, T);
			visit1[x][y] = 0;	
		} 		
	}		
	else if(map1[x][y] == '*')
		return ;
	else if(map1[x][y] == '.') {
		visit1[x][y] = 1;
		for(int i = 0; i < 4; ++i) {
			if(x + dirx[i] < 0 || x + dirx[i] >= n || y + diry[i] < 0 || y + diry[i] >= m)
				continue;
			else {
				if(visit1[x + dirx[i]][y + diry[i]] == 0) {
					DFS1(x + dirx[i], y + diry[i], T + 1);
				}
			}		
		}
		visit1[x][y] = 0;	
	}
}


int main() {
	int c;
	cin >> c;
	while(c--) {
		cin >> n >> m >> t;
		mint = 2 * t;
		for(int i = 0; i < n; ++i)
			for(int j = 0; j < m; ++j) {
				cin >> map1[i][j];
				if(map1[i][j] == '*')
					visit1[i][j] = 1;
				else
					visit1[i][j] = 0;
			}
		for(int i = 0; i < n; ++i)
			for(int j = 0; j < m; ++j) {
				cin >> map2[i][j];
				if(map2[i][j] == '*')
					visit2[i][j] = 1;
				else
					visit2[i][j] = 0;
			}
		map1[0][0] = '.';
		DFS1(0, 0, 0);
		if(mint <= t)
			cout << "YES" << endl;
		else
			cout << "NO" << endl;
	}
	return 0;
}

正确代码

#include
#include
#include
using namespace std;

int c, m, n, t, door, x2, y2, t1;
bool yy;
char map_1[12][12];
char map_2[12][12];
bool vis_1[12][12];
bool vis_2[12][12];
int move_a[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
void dfs_1(int a, int b, int num);
void dfs_2(int a, int b, int num);

void dfs_1(int a, int b, int num)
{
    if(yy || num > t)return ;
    if(map_1[a][b] == 'P')
    {
        if(num <= t)yy = 1;
        return ;
    }
    int ans = t - num - abs(a - x2) - abs(b - y2);
    if(door == 1 && ans < 0)return ;
    if(!yy)
    {
        for(int i = 0; i < 4; i++)
        {
            int x = a + move_a[i][0];
            int y = b + move_a[i][1];
            if(0 <= x && x < n && 0 <= y && y < m && !vis_1[x][y])
            {
                vis_1[x][y] = 1;
                if(map_1[x][y] == '#')
                {
                    if(map_2[x][y] != '#' && map_2[x][y] != '*')
                        dfs_2(x, y, num + 1);
                }
                else dfs_1(x, y, num + 1);
                if(yy)return ;
                vis_1[x][y] = 0;
            }
        }
    }
    return ;
}

void dfs_2(int a, int b, int num)
{
    if(yy || num > t)return ;
    if(map_2[a][b] == 'P')
    {
        if(num <= t)yy = 1;
        return ;
    }
    int ans = t - num - abs(a - x2) - abs(b - y2);
    if(door == 2 && ans < 0)return ;
    if(!yy)
    {
        for(int i = 0; i < 4; i++)
        {
            int x = a + move_a[i][0];
            int y = b + move_a[i][1];
            if(0 <= x && x < n && 0 <= y && y < m && !vis_2[x][y])
            {
                vis_2[x][y] = 1;
                if(map_2[x][y] == '#')
                {
                    if(map_1[x][y] != '#' && map_1[x][y] != '*')
                        dfs_1(x, y, num + 1);
                }
                else dfs_2(x, y, num + 1);
                if(yy)return ;
                vis_2[x][y] = 0;
            }
        }
    }
    return ;
}

int main()
{
    scanf("%d", &t1);
    while(t1--)
    {
        yy = false;
        memset(vis_1, 0, sizeof vis_1);
        memset(vis_2, 0, sizeof vis_2);
        scanf("%d%d%d", &n, &m, &t);
        for(int i = 0; i < n; i++)
        {
            scanf("%s", map_1[i]);
            for(int j = 0; j < m; j++)
            {
                if(map_1[i][j] == 'P')door = 1, x2 = i, y2 = j;
                else if(map_1[i][j] == '*')vis_1[i][j] = 1;
            }
        }
        for(int i = 0; i < n; i++)
        {
            scanf("%s", map_2[i]);
            for(int j = 0; j < m; j++)
            {
                if(map_2[i][j] == 'P')door = 2, x2 = i, y2 = j;
                else if(map_2[i][j] == '*')vis_2[i][j] = 1;
            }
        }
        vis_1[0][0] = 1;
        dfs_1(0, 0, 0);
        if(yy)printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}

注意点

1.公主可以在任意一层
2.#不能在两层的同一位置
3.从第二层还可以回到第一层