路径之谜 第七届蓝桥杯决赛 java

路径之谜

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

解题思路：暴力DFS

package GSZT2017;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class DFSLuJing {
	static int n;
	static int[] a=new int[25];
	static int[] b=new int[25];
	static int[] ta=new int[25];
	static int[] tb=new int[25];
	static int[][] vis=new int[25][25];
	static int[] sum=new int [450];
	static boolean flag=true;
	static int[][] dir= {{0,1},{0,-1},{1,0},{-1,0}};

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
		String s=bf.readLine();
		n=Integer.parseInt(s);
		s=bf.readLine();
		for (int i = 0; i < n; i++) {
			String[] ac=s.split(" ");
			a[i]=Integer.parseInt(ac[i]);
		}
		s=bf.readLine();
		for (int i = 0; i < n; i++) {
			String[] ac=s.split(" ");
			b[i]=Integer.parseInt(ac[i]);
		}
		ta[0]++;
		tb[0]++;
//		vis[0][0]=1;
		dfs(0,0,1);

	}

	public static void dfs(int x,int y,int k) {
		//如果flag等于0说明路径前面存在不满足，可直接返回
		if(!flag) return;
		//如果是最后一个判断输出
		if(x==n-1&&y==n-1) {
			boolean f=true;//这里如果不重新定义用flag则flag的值将永远为false
			//对结果进行判定，如果满足，则输出，否则不输出
			for (int i = 0; i < a.length; i++) {
				if(ta[i]!=a[i]||tb[i]!=b[i]) {
					f=false;
					break;
				}
			}
			if(f) {
				for (int i = 0; i < k; i++) {
					System.out.print(sum[i]+" ");
				}
			}
			return;
		}
		//dfs 这里的4代表四个方向可以走
		for (int i = 0; i <4 ; i++) {
			int nextx=x+dir[i][0];
			int nexty=y+dir[i][1];
			//符合条件则进行dfs,可以直接加入最终结果即可进入。
			if(nextx>=0&&nextx<n&&nexty>=0&&nexty<n&&ta[nextx]<a[nextx]&&tb[nexty]<b[nexty]&&vis[nextx][nexty]==0) {
				ta[nextx]++;
				tb[nexty]++;
				vis[nextx][nexty]=1;
				sum[k]=nextx+nexty*n;
				dfs(nextx,nexty,k+1);
				//回溯
				ta[nextx]--;
				tb[nexty]--;
				vis[nextx][nexty]=0;
			}
		}
	}

}