蓝桥杯——玩具蛇

题目

小蓝有—条玩具蛇，一共有16节，上面标着数字1至16。每—节都是一个正方形的形状。相邻的两节可以成直线或者成90度角。
小蓝还有一个4×4的方格盒子，用于存放玩具蛇，盒子的方格上依次标着字母A到Р共16个字母。
小蓝可以折叠自己的玩具蛇放到盒子里面。他发现，有很多种方案可以将玩具蛇放进去。
下图给出了两种方案:

请帮小蓝计算一下，总共有多少种不同的方案。如果两个方案中，存在玩具蛇的某一节放在了盒子的不同格子里，则认为是不同的方案。

分析

很明显的DFS，只需要定义方向，然后注意不要越界，每次放置好一段之后需要标记，标记完之后再接着搜索，搜完之后要删除标记。

代码

num = [[0] * 4 for i in range(4)]
direction = [[1, 0], [-1, 0], [0, 1], [0, -1]]
ans = 0
def dfs(x, y, count):
    global ans
    if count == 16:
        ans += 1
        return
    for i in range(4):
        dx = x + direction[i][0]
        dy = y + direction[i][-1]
        if 0 <= dx < 4 and 0 <= dy < 4 and num[dx][dy] == 0:
            num[dx][dy] = 1
            dfs(dx, dy, count+1)
            num[dx][dy] = 0
for i in range(4):
    for j in range(4):
        num[i][j] = 1
        dfs(i, j, 1)
        num[i][j] = 0
print(ans)