DFS解决路径之迷（蓝桥杯练习）

小明冒充X星球的骑士，进入了一个奇怪的城堡。城堡里边什么都没有，只有方形石头铺成的地面。
假设城堡地面是 n x n 个方格。
按习俗，骑士要从西北角走到东南角。可以横向或纵向移动，但不能斜着走，也不能跳跃。
每走到一个新方格，就要向正北方和正西方各射一箭。（城堡的西墙和北墙内各有 n 个靶子）
同一个方格只允许经过一次。但不必走完所有的方格。
如果只给出靶子上箭的数目，你能推断出骑士的行走路线吗？
有时是可以的，比如图中的例子。
本题的要求就是已知箭靶数字，求骑士的行走路径（测试数据保证路径唯一）

输入：
第一行一个整数N(0<N<20)，表示地面有 N x N 个方格
第二行N个整数，空格分开，表示北边的箭靶上的数字（自西向东）
第三行N个整数，空格分开，表示西边的箭靶上的数字（自北向南） 

输出： 
一行若干个整数，表示骑士路径
为了方便表示，我们约定每个小格子用一个数字代表，从西北角开始编号: 0,1,2,3....
比如，上图中的方块编号为：
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

源代码：

#include<iostream>
#include<vector>
using namespace std;
int n;
struct position
{
    int h;
    int z;
};
vector<position> res;
vector<int>ans;
int north[30];
int west[30];
int flag[22][22] = { 0 };
int a[4] = { 0,0,-1,1 };
int b[4] = { 1,-1,0,0 };
int check(int x, int y)
{

    if (x == n && y == n)
    {
        for (int i = 1; i <= n; i++)
        {
            if (north[i] != 0 || west[i] != 0)
                return 0;
        }
        for (int i = 0; i < res.size(); i++)
        {
            int x1 = res[i].h;
            int y1 = res[i].z;
            int sum = n * (y1 - 1) + x1 - 1;
            ans.push_back(sum);
        }
        return 0;
    }
    return 1;
}
int pd(int x, int y)
{
    if (flag[x][y] == 1)
        return 0;
    if (x<1 || x>n || y > n || y < 1 || north[x] <= 0 || west[y] <= 0)
        return 0;
    return 1;
}
void dfs(int x, int y)
{
    if (check(x, y) == 0)
        return;
    int x1, y1;
    for (int i = 0; i <= 3; i++)
    {
        x1 = a[i] + x;
        y1 = b[i] + y;
        if (pd(x1, y1))
        {
            flag[x1][y1] = 1;
            north[x1]--;
            west[y1]--;
            res.push_back({ x1,y1 });
            dfs(x1, y1);
            flag[x1][y1] = 0;
            res.pop_back();
            north[x1]++;
            west[y1]++;
        }
    }
}
int main()
{
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        cin >> north[i];
    }
    for (int i = 1; i <= n; i++)
    {
        cin >> west[i];
    }
    flag[1][1] = 1;
    res.push_back({ 1,1 });
    north[1]--;
    west[1]--;
    dfs(1, 1);
    for (int i = 0; i < ans.size(); i++)
    {
        printf("%d ", ans[i]);
    }
    return 0;
}

这里我用了两个vector，一个用来存放正确路径经过点的坐标，一个用来存放相应坐标对应输出的结果。

感觉这题真的很难，思路大体上都有，但是上手就会很困难，并且许多细节会遗漏。这里check函数里自认为是最难想的地方：在从左上角走到右下角后，用check判断各个靶子上的箭是否为0。而因为要输出路径，所以也要在用check函数判断通过正确的路径走到终点后，将走过的路径坐标保存在vector中。

PS:在判断函数中，某方向箭拔完也要return 0。因为在DFS函数中，先进行check操作（即是否走到终点），再进行判定操作，所以在进行判定操作时，一定是没有走到终点的，所以如果此时某方向箭拔完了，就无法走到终点。