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

题目描述

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

思路：如果直接使用深搜会超时。所以在深搜的时候用箭靶数量作为限制条件加入。每走一步就减去相应的箭靶数量当走到出口并且箭靶数量全都为0的时候就是我们要找到的路径。

import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class Main {
    static int N;
    static int[][] map;
    static List<Integer> numList = new ArrayList<>(); // 我这里用list保存走过的格子，因为之前用数组保存的时候还原状态会出问题。

    static void f(int n[], int[] m, int i, int j) {
        n[i] -= 1; // 把当前各自的箭靶数量减去
        m[j] -= 1;
        numList.add(map[j][i]); // 当前格子加到已经走过的路径
        map[j][i] = -1; 当前格子设置为-1，这样就不会走重复的路了
        if (i == N - 1 && j == N - 1) { // 判断是否走到终点
            int sum = 0;
            for (int k = 0; k < N; k++) {// 把所有的箭靶数量加起来
                sum+=n[k]+m[k];
            }
            if (sum== 0){ 
                for (int k : numList) {
                    System.out.print(k + " ");
                }
                System.out.println(" ");
            }
            map[j][i] = j * N + i;// 以防走到终点但是箭靶数量不对，所以还是要还原
            numList.remove(numList.size() - 1);
            n[i] += 1;
            m[j] += 1;
            return;
        }

        if (i + 1 < N && map[j][i + 1] != -1 && n[i + 1] - 1 >= 0 && m[j] - 1 >= 0) {// 判断能不能往右走
            f(n, m, i + 1, j);
        }
        if (j + 1 < N && map[j + 1][i] != -1 && n[i] - 1 >= 0 && m[j + 1] - 1 >= 0) {// 判断能不能往下走
            f(n, m, i, j + 1);
        }
        if (i - 1 >= 0 && map[j][i - 1] != -1 && n[i - 1] - 1 >= 0 && m[j] - 1 >= 0) {// 判断能不能往左走
            f(n, m, i - 1, j);
        }
        if (j - 1 >= 0 && map[j - 1][i] != -1 && n[i] - 1 >= 0 && m[j - 1] - 1 >= 0) {// 判断能不能往上走
            f(n, m, i, j - 1);
        }
        map[j][i] = j * N + i;// 还原状态
        numList.remove(numList.size() - 1);
        n[i] += 1;
        m[j] += 1;
    }

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        N = in.nextInt();
        map = new int[N][N];// 初始化我们要走的格子里的数字
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                map[j][i] = j * N + i;
            }
        }
        int[] n = new int[N];  // x轴 数组的第二维
        for (int i = 0; i < N; i++) {
            n[i] = in.nextInt();
        }
        int[] m = new int[N]; // y 轴
        for (int i = 0; i < N; i++) {
            m[i] = in.nextInt();
        }

        f(n, m, 0, 0);
    }
}