问题描述
小明冒充X星球的骑士,进入了一个奇怪的城堡。
城堡里边什么都没有,只有方形石头铺成的地面。
假设城堡地面是 n x n 个方格。如下图所示:按习俗,骑士要从西北角走到东南角。
可以横向或纵向移动,但不能斜着走,也不能跳跃。
每走到一个新方格,就要向正北方和正西方各射一箭。
(城堡的西墙和北墙内各有 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
题目分析
一道很基础的dfs的题目
具体操作就是走到一个格子上就进行射箭并走下一步 若不满足条件则进行回溯
具体代码
#include<iostream>
#include<stack>
using namespace std;
const int N = 25;
int a[N];
int b[N];
int mp[N][N];//记录整张图的状态
int n;
int len[4][2] = { {1,0},{-1,0},{0,1},{0,-1} };
stack<int>stk;//记录走过的点
//判断到达终点后是否满足要求
bool check() {
for (int i = 0;i < n;i++) {
if (a[i] || b[i]) {
return false;
}
}
return true;
}
bool dfs(int x,int y) {
if (x == n - 1 && y == n - 1) {
if (check()) {
return true;
}
return false;
}
for (int i = 0;i < 4;i++) {
int xx = x + len[i][0];
int yy = y + len[i][1];
//判断合理性
if (xx >= 0 && yy >= 0 && xx < n && yy < n && mp[xx][yy] == 0) {
if (a[xx] && b[yy])a[xx]--, b[yy]--;//判断走到该格能否满足要求
else continue;
mp[xx][yy] = 1;
stk.push(xx + yy * n);//此处记录的可以理解为当前格子的编号
if (dfs(xx, yy))return true;
//回溯
mp[xx][yy] = 0, a[xx]++, b[yy]++, stk.pop();
}
}
return false;
}
void Print() {
stack<int>ans;
while (!stk.empty()) {
ans.push(stk.top());
stk.pop();
}
while (!ans.empty()) {
cout << ans.top() << " ";
ans.pop();
}
}
int main() {
cin >> n;
for (int i = 0;i < n;i++)cin >> a[i];
for (int i = 0;i < n;i++)cin >> b[i];
mp[0][0] = 1;
stk.push(0);
a[0]--;
b[0]--;
if (dfs(0, 0)) {
Print();
}
}