蓝桥杯 第七届javaB组决赛 第四题 路径之谜

题目

路径之谜

小明冒充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，否则按无效代码处理。

思路

主要就是深搜+回溯，听大佬说用网络流但是不会在此就只好这样写啦。
题意可知小明每走一步都会往北往西射一只箭。解题的时候我们每走一步就可以将北边和西边的剪拔下一只。
要注意在每走一步之前要判断下一步的两个箭靶上是否有箭。以及到达终点后要判断所有箭靶上面是否都没有箭。满足这两个条件就可以输出path了。

代码

import java.io.BufferedInputStream;
import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Scanner;

import javax.xml.crypto.Data;

public class Main{
	static int n;//数组大小
	static int north[]=null,west[]=null;//存储北边和西边的箭靶
	static int vis[][]=null;//标记数组
	static int dir[][]={{1,0},{-1,0},{0,1},{0,-1}};//方向
	public static void main(String[] args) throws Exception{
		Scanner sc=new Scanner(new BufferedInputStream(System.in));
		n=sc.nextInt();
		north=new int[n];
		west=new int[n];
		vis=new int [n][n];
		for(int i=0;i<n;i++) {
			north[i]=sc.nextInt();
		}
		for(int i=0;i<n;i++) {
			west[i]=sc.nextInt();
		}
		north[0]--;
		west[0]--;
		vis[0][0]=1;
		dfs(0,0,"0");//从起点开始深搜，注意要将起点的北边和西边的箭靶减1
	}

	public static void dfs(int x,int y,String path)
	{
		if(x==n-1&&y==n-1){//到达终点
			int cnt=0;
			for(int i=0;i<n;i++){
				if(west[i]==0) cnt++;
				if(north[i]==0) cnt++;
			}
			if(cnt==n*2)//如果所有箭靶上的箭均为0
				System.out.println(path);//输出路径
			return ;
		}
		else{
			for(int i=0;i<4;i++)//四个方向
			{
				int tx=x+dir[i][0];
				int ty=y+dir[i][1];
				if(tx>=0&&tx<n&&ty>=0&&ty<n)//坐标越界判断
					if(north[ty]>0&&west[tx]>0&&vis[tx][ty]==0)//箭靶上箭是否足够和是否已经访问过
					{
						north[ty]--;
						west[tx]--;
						String newpath=path+" "+String.valueOf(tx*n+ty);
						vis[tx][ty]=1;
						dfs(tx,ty,newpath);//继续深搜
						north[ty]++;
						west[tx]++;
						vis[tx][ty]=0;//回溯后要还原箭靶和标记数组
					}
			}
		}
	}
}