路径之谜
小明冒充X星球的骑士,进入了一个奇怪的城堡。
城堡里边什么都没有,只有方形石头铺成的地面。
假设城堡地面是 n x n 个方格。【如图1.png】所示。
按习俗,骑士要从西北角走到东南角。
可以横向或纵向移动,但不能斜着走,也不能跳跃。
每走到一个新方格,就要向正北方和正西方各射一箭。
(城堡的西墙和北墙内各有 n 个靶子)
同一个方格只允许经过一次。但不必做完所有的方格。
如果只给出靶子上箭的数目,你能推断出骑士的行走路线吗?
有时是可以的,比如图1.png中的例子。
本题的要求就是已知箭靶数字,求骑士的行走路径(测试数据保证路径唯一)
输入:
第一行一个整数N(0<N<20),表示地面有 N x N 个方格
第二行N个整数,空格分开,表示北边的箭靶上的数字(自西向东)
第三行N个整数,空格分开,表示西边的箭靶上的数字(自北向南)
输出:
一行若干个整数,表示骑士路径。
为了方便表示,我们约定每个小格子用一个数字代表,从西北角开始编号: 0,1,2,3…
比如,图1.png中的方块编号为:
0 1 2 3
4 5 6 7
8 9 10 11
12 13 14 15
示例:
用户输入:
4
2 4 3 4
4 3 3 3
程序应该输出:
0 4 5 1 2 3 7 11 10 9 13 14 15
资源约定:
峰值内存消耗 < 256M
CPU消耗 < 1000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
注意:不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意:主类的名字必须是:Main,否则按无效代码处理。
Ideas
这道题类似一个迷宫问题,迷宫问题想搜索,最优问题想广搜,不然想深搜。
对于深搜函数DFS来说,肯定需要传入一个位置坐标x和y,然后这道题还需要输出骑士走的路径,所以在最后找到一个符合条件的路径之后还要输出。
之后就是正常的DFS套路了,四个方向都走一走,保证不要越界,并且还需要一个visit数组表示某个位置是否被访问过,最后从西北角搜索到东南角就OK了。
当然这道题还需要确保箭靶数字符合要求,所以我们其实可以利用这个条件做一个剪枝,当我们走到一个位置之后,其实可以把对应位置的两个箭靶数字减去1,如果某一个位置箭靶数字剪成了负数,说明白不符合条件,就可以直接return。
Code
C++
#include <iostream>
#include <vector>
using namespace std;
struct Node {
bool flag;
int x, y;
};
Node map[20][20];
vector<int> road;
int N, X[20], Y[20], sum;
int dir[4][2] = {{0, 1},
{1, 0},
{-1, 0},
{0, -1}};
bool dfs(int x, int y) {
if (x == N - 1 && y == N - 1) {
for (int i = 0; i < N; i++) {
if (X[i] || Y[i] || sum)
return false;
road.push_back(x * N + y);
return true;
}
}
road.push_back(x * N + y);
map[x][y].flag = true;
for (int i = 0; i < 4; i++) {
int tx = x + dir[i][0];
int ty = y + dir[i][1];
if (tx < 0 || tx > (N - 1) || ty < 0 || ty > (N - 1))
continue;
if (!map[tx][ty].flag && (X[tx] > 0 && Y[ty] > 0)) {
X[tx]--;Y[ty]--;sum -= 2;
if (dfs(tx, ty))
return true;
else {
X[tx]++;
Y[ty]++;
sum += 2;
}
}
}
map[x][y].flag = false;
road.erase(road.begin() + road.size() - 1);
return false;
}
int main() {
cin >> N;
for (int i = 0; i < N; i++){
cin >> Y[i];
sum += Y[i];
}
for (int i = 0; i < N; i++) {
cin >> X[i];
sum += X[i];
}
for (int i = 0; i < N; i++) {
for (int j = 0; j < N; j++) {
map[i][j].flag = false;
map[i][j].x = i;
map[i][j].y = j;
}
}
X[0]--;Y[0]--;sum -= 2;
dfs(0, 0);
for (int i = 0; i < road.size(); i++)
cout << road[i] << ' ';
return 0;
}
Python
def dfs(r, c):
"""
Args:
r: 当前搜索行
c: 当前搜索列
Returns: Bool值,表示是否找到了一条符合条件路径
"""
if r == n - 1 and c == n - 1:
if sum(north_target) > 0 or sum(west_target) > 0: # 如果最后还有箭靶上面的数目不为零,也不符合要求
return False
path.append(r * n + c)
print(' '.join(map(str, path)))
return True
path.append(r * n + c)
direction = [(0, 1), (1, 0), (0, -1), (-1, 0)]
for dr, dc in direction:
nr, nc = r + dr, c + dc # 下一个搜索位置的行、列
# 保证下一个搜索的位置符合要求
if -1 < nr < n and -1 < nc < n and not visit[nr][nc] and north_target[nc] > 0 and west_target[nr] > 0:
visit[nr][nc] = True
north_target[nc] -= 1
west_target[nr] -= 1
if dfs(nr, nc):
return True
else:
visit[nr][nc] = False
north_target[nc] += 1
west_target[nr] += 1
path.pop()
return False
if __name__ == '__main__':
n = int(input())
north_target = list(map(int, input().split()))
west_target = list(map(int, input().split()))
path = []
visit = [[False] * n for _ in range(n)]
visit[0][0] = True
north_target[0] -= 1
west_target[0] -= 1
dfs(0, 0)
在线评测:https://www.acwing.com/problem/content/description/3193/
这道题至今只有1个人AC了,上面的代码应该只能拿70分,求AC大佬分享。