题目描述
小明冒充 X 星球的骑士,进入了一个奇怪的城堡。
城堡里边什么都没有,只有方形石头铺成的地面。
假设城堡地面是 n×n 个方格。如下图所示。
按习俗,骑士要从西北角走到东南角。可以横向或纵向移动,但不能斜着走,也不能跳跃。每走到一个新方格,就要向正北方和正西方各射一箭。(城堡的西墙和北墙内各有 n个靶子)同一个方格只允许经过一次。但不必走完所有的方格。如果只给出靶子上箭的数目,你能推断出骑士的行走路线吗?有时是可以的,比如上图中的例子。
本题的要求就是已知箭靶数字,求骑士的行走路径(测试数据保证路径唯一)
输入
第一行一个整数 N (0≤N≤20),表示地面有 N ×N 个方格。
第二行 N 个整数,空格分开,表示北边的箭靶上的数字(自西向东)
第三行 N 个整数,空格分开,表示西边的箭靶上的数字(自北向南)
输出
输出一行若干个整数,表示骑士路径。
为了方便表示,我们约定每个小格子用一个数字代表,从西北角开始编号: 0,1,2,3
0 1 2 3
4 5 6 7
8 9 10 11
12 13 14 15
用例输入 1
4
2 4 3 4
4 3 3 3
用例输出 1
0 4 5 1 2 3 7 11 10 9 13 14 15
思路分析
DFS问题都是有套路的,一般来说就这几个步骤:
1.确定该的题目的边界状态
2.找到转移的方法
3.找到问题的出口,计数或某种状态
这是伪代码:
int check(参数)
{
if(满足条件)
return 1;
return 0;
}
bool pd(参数){
相应操作
}
void dfs(int step)
{
判断边界pd()
{
不在边界内,即回溯
}
尝试每一种可能
{
满足check条件
标记
继续下一步dfs(step+1)
恢复初始状态(回溯的时候要用到)
}
}
而对于这个问题,我们可以这样:
我们可以采用逆推的方式,我们从[0][0]的位置开始走,每次可以向上下左右四个方向走,每走一步就拔掉一支对应的北边和西边箭靶上的箭,如果走到出口的时候正好靶子上的箭都没了,因为路径唯一,证明我们找到正确路径。
#include<bits/stdc++.h>
using namespace std;
const int N=30;//题目给出n最大20,我直接设置为30
int n;//行列数
int row[N];//用于存放行靶数
int col[N];//用于存放列靶数
int dx[]={-1,0,1,0};//横坐标偏移量(上右下左)
int dy[]={0,1,0,-1};//纵坐标偏移量(上右下左)
vector<int> path;//用来存放路线(即经过的点对应的值)
bool st[N][N];//用于标记是否走过
bool dfs(int x,int y){
//判断是否已经走到底了 ,在看是否走对了
if(x==n-1&&y==n-1){
for(int i=0;i<n;i++){
if(row[i]!=0||col[i]!=0){
return false;
}
}
return true;
}
//上右下左试着走
for(int i=0;i<4;i++){
int a=x+dx[i];int b=y+dy[i];
//判断这一步能不能走
if(a<0||a>=n||b<0||b>=n||st[a][b])continue;//超出边界或已经走过
if(row[a]<=0||col[b]<=0)continue;//靶子上的箭已经拔光
//能走
st[a][b]=true;//这点标志为已走
row[a]--;
col[b]--;
path.push_back(a*n+b);
if(dfs(a,b)) return true;//已经走到尽头,并且无误
//走错了,回溯到上一个位置
st[a][b]=false;
row[a]++;
col[b]++;
path.pop_back();//删除末尾
}
return false;
}
int main(){
//输入
cin>>n;
for(int i=0;i<n;i++){
cin>>col[i];
}
for(int i=0;i<n;i++){
cin>>row[i];
}
//从第一个位置开始走的,靶数减一;标志为已经走过
row[0]--;
col[0]--;
st[0][0]=true;
path.push_back(0);
//开始搜索
dfs(0,0);//每次从第一个位置开始走
//输出结果
int num=path.size();
for(int i=0;i<num;i++)
cout<<path[i]<<' ';
return 0;
}