[蓝桥杯][2016年第七届真题]路径之谜题解

题目描述

假设城堡地面是 n x n 个方格。

按习俗，骑士要从西北角走到东南角。
可以横向或纵向移动，但不能斜着走，也不能跳跃。
每走到一个新方格，就要向正北方和正西方各射一箭。
（城堡的西墙和北墙内各有 n 个靶子）

同一个方格只允许经过一次。但不必走完所有的方格。

如果只给出靶子上箭的数目，你能推断出骑士的行走路线吗？

有时是可以的，比如图1中的例子。

本题的要求就是已知箭靶数字，求骑士的行走路径（测试数据保证路径唯一）

输入

第一行一个整数N(0<N<20)，表示地面有 N x N 个方格
第二行N个整数，空格分开，表示北边的箭靶上的数字（自西向东）
第三行N个整数，空格分开，表示西边的箭靶上的数字（自北向南）

输出

一行若干个整数，表示骑士路径。

为了方便表示，我们约定每个小格子用一个数字代表，从西北角开始编号: 0,1,2,3…
比如，图1中的方块编号为：

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即可，注意处理好回溯时的一些状态；

1、设置一个二维数组，数值为0，1，…，即为（方块）坐标(x,y)代表的数字；
2、再设置两个一维数组，north，west；
3、设置一个vector，保存路径上的点；
4、设置vis数组，坐标是否已访问过；

每次深搜到点(x,y)，将点加入路径，将north[y],west[x]减一，直到(x,y)=(n-1,n-1)并且north，west数组为0，这时找出路径，输出路径结束。若没结束，则分别在(x,y)上下左右深搜。处理完此点的四周，回溯时注意要把状态复原，即north,west,path,vis变回以前。

代码

#include<bits/stdc++.h>
using namespace std;

const int maxn=22;
int n;
char Map[maxn][maxn];
int north[maxn],west[maxn];
bool vis[maxn][maxn];
int X[]= {-1,1,0,0};//四个方向
int Y[]= {0,0,-1,1};
vector<int>path;

void dfs(int x,int y) { //走到(x,y)
	north[y]-=1;
	west[x]-=1;
	path.push_back(Map[x][y]);
	vis[x][y]=true;
	if(x==n-1&&y==n-1) {
		bool flag=false;
		for(int i=0; i<n; i++) {
			if(north[i]!=0||west[i]!=0) {
				flag=true;
				break;
			}
		}
		if(flag==false) {
			for(int i=0; i<path.size(); i++) {
				printf("%d ",path[i]);
			}
		}
	} else {
		for(int i=0; i<4; i++) {
			int newX=x+X[i];
			int newY=y+Y[i];
			if(newX>=0 && newX<n && newY>=0 && newY<n && !vis[newX][newY]) {
				if(north[newY]>0&&west[newX]>0) {
					dfs(newX,newY);
				}
			}
		}
	}
	//回溯
	vis[x][y]=false;
	path.pop_back();
	north[y]+=1;
	west[x]+=1;
}

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 a=0;
	for(int i=0; i<n; i++) {
		for(int j=0; j<n; j++) {
			Map[i][j]=a++;
		}
	}
	dfs(0,0);
	return 0;
}