蓝桥训练营-路径之谜-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

思路：深度优先搜索，记录每次深搜的横纵坐标再对对应方向的靶子数组进行计数操作，最后满足到达终点，且对靶子数的叠加刚好都对应即可输出路径

Code：

import java.util.Scanner;
import java.util.*;


public class Main{

    static  int[][] dir={{0,1},{1,0},{0,-1},{-1,0}};
    static int[] west;
    static int[] north;
    static int[] visit;
    static int n;
    static ArrayList<Integer> list=new ArrayList<>();
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        n = scanner.nextInt();
        visit=new int[n*n];
        north=new int[n];
        west=new int[n];
        for (int i = 0; i <n; i++) {
            north[i]= scanner.nextInt();
        }
        for (int i = 0; i <n; i++) {
            west[i]= scanner.nextInt();
        }
        list.add(0);
        north[0]--;
        west[0]--;
        visit[0]=1;
        dfs(0,0,0);
    }

    private static void dfs(int x, int y, int num) {

        if (x*n+y==n*n-1){
            for (int i = 0; i < n; i++) {
                if (north[i]!=0||west[i]!=0){
                    return;
                }
            }
            for (int i = 0; i <=num; i++) {
                System.out.print(list.get(i)+" ");
            }
            return;
        }
        for (int i = 0; i < dir.length; i++) {
            int x1=x+dir[i][0];
            int y1=y+dir[i][1];
            if (x1>=0&&x1<n&&y1>=0&&y1<n&&visit[x1*n+y1]!=1&&west[x1]>0&&north[y1]>0){
                visit[x1*n+y1]=1;
                north[y1]--;
                west[x1]--;
                list.add(x1*n+y1);
                dfs(x1,y1,num+1);
                list.remove(list.size()-1);
                north[y1]++;
                west[x1]++;
                visit[x1*n+y1]=0;
            }
        }
    }
}