DFS|路径之谜

路径之谜

小明冒充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

import java.util.Scanner;

public class 路径 {
	static int[] dx = { 0, 1, -1, 0 };
	static int[] dy = { 1, 0, 0, -1 };
	static int[] path = new int[405];// 装节点数字
	static int[] north = new int[25];
	static int[] west = new int[25];
	static int[][] vis = new int[25][25];
	static int[][] maze = new int[25][25];
	static int n, m;

	static void init() {
		Scanner sc = new Scanner(System.in);
		n = sc.nextInt();
		int temp = 0;
		m = 0;
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < n; j++) {
				maze[i][j] = temp++;
				vis[i][j] = -1;
			}
		}
		for (int i = 0; i < n; i++) {
			north[i] = sc.nextInt();
			m += north[i];
		}
		for (int i = 0; i < n; i++) {
			west[i] = sc.nextInt();
		}
	}

	private static void dfs(int cur, int x, int y, int gx, int gy) {
		if (cur == m) {
			if (checkPath()) {// 检查路径是否到达终点，且符合箭靶条件
				printPath();
			}
			return;
		}

		for (int i = 0; i < 4; i++) {
			int nx = x + dx[i], ny = y + dy[i];
			if (judge(nx, ny)) {
				vis[nx][ny] = 1;
				path[cur] = maze[nx][ny];
				dfs(cur + 1, nx, ny, gx, gy);
				vis[nx][ny] = -1;
			}
		}
	}

	private static boolean checkPath() {
		if (path[m - 1] == maze[n - 1][n - 1]) {
			int[] check = new int[25];
			int[] check2 = new int[25];
			for (int i = 0; i < n; i++) {
				for (int j = 0; j < n; j++) {
					if (vis[j][i] == 1)
						check2[i]++;
					if (vis[i][j] == 1)
						check[i]++;
				}
			}
			for (int i = 0; i < n; i++) {
				if (!(check[i] == west[i] && check2[i] == north[i]))
					return false;
			}
			return true;
		}
		return false;
	}

	private static boolean judge(int nx, int ny) {
		if ((0 <= nx && nx < n) && (0 <= ny && ny < n) && (vis[nx][ny] == -1)) {
			// 剪枝
			int sum = 0;
			for (int i = 0; i < n; i++) {
				if (vis[nx][i] == 1)
					sum++;
			}
			if (sum >= west[nx])
				return false;
			sum = 0;
			for (int i = 0; i < n; i++) {
				if (vis[i][ny] == 1)
					sum++;
			}
			if (sum >= north[ny])
				return false;
			return true;
		}
		return false;
	}

	private static void printPath() {
		for (int i = 0; i < m; i++) {
			if (i != 0)
				System.out.print(" ");
			System.out.print(path[i]);
		}
		System.out.println();
	}

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		init();
		vis[0][0] = 1;
		path[0] = maze[0][0];
		dfs(1, 0, 0, n - 1, n - 1);
	}

}