小明冒充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
思路
在上方的箭靶数(north数组)就是走过该列的格子数,在左方的箭靶数(west数组)就是走过该行的格子数。
直接dfs走到终点,用mark数组标记该点是否走过。
最后用cntX数组计算该行走过的格子数,cntY数组计算该列走过的格子数
最后判断cntXwest,cntYnorth
#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
int north[25],west[20],mp[25][25],mark[25][25];
int n;
int d[4][2]={{0,1},{0,-1},{1,0},{-1,0}};
vector<int>way;
void dfs(int x,int y)
{
if(x==n-1&&y==n-1)//如果到达终点
{
int cntX[25],cntY[25];
memset(cntX,0,sizeof(cntX));
memset(cntY,0,sizeof(cntY));
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
{
cntX[i]+=mark[i][j];
cntY[j]+=mark[i][j];
}
int flag=true;
for(int i=0;i<n;i++)
{
if(cntX[i]!=west[i])
{
flag=false;
break;
}
}
for(int i=0;i<n;i++)
{
if(cntY[i]!=north[i])
{
flag=false;
break;
}
}
if(flag)
{
for(int i=0;i<way.size();i++)
{
if(i==0)
cout << way[i];
else
cout << " " << way[i];
}
}
}
int tx,ty;
for(int i=0;i<4;i++)
{
tx=x+d[i][0];
ty=y+d[i][1];
if(tx>=0&&tx<n&&ty>=0&&ty<n&&mark[tx][ty]==0)
{
mark[tx][ty]=1;
way.push_back(mp[tx][ty]);
dfs(tx,ty);
mark[tx][ty]=0;
way.pop_back();
}
}
}
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&north[i]);
for(int i=0;i<n;i++)
scanf("%d",&west[i]);
int t=0;
for(int i=0;i<n;i++)
for(int j=0;j<n;j++)
mp[i][j]=t++;//给格子编码
mark[0][0]=1;
way.push_back(0);
dfs(0,0);
}