蓝桥杯Python历届真题02-兰顿蚂蚁（dfs深度优先搜索，超详细）

时间限制: 1Sec 内存限制: 128MB

问题描述：

兰顿蚂蚁，是于1986年，由克里斯·兰顿提出来的，属于细胞自动机的一种。
平面上的正方形格子被填上黑色或白色。在其中一格正方形内有一只“蚂蚁”。
蚂蚁的头部朝向为：上下左右其中一方。
蚂蚁的移动规则十分简单：
若蚂蚁在黑格，右转90度，将该格改为白格，并向前移一格；
若蚂蚁在白格，左转90度，将该格改为黑格，并向前移一格。
规则虽然简单，蚂蚁的行为却十分复杂。刚刚开始时留下的路线都会有接近对称，像是会重复，但不论起始状态如何，蚂蚁经过漫长的混乱活动后，会开辟出一条规则的“高速公路”。
蚂蚁的路线是很难事先预测的。
你的任务是根据初始状态，用计算机模拟兰顿蚂蚁在第n步行走后所处的位置。

输入格式：

输入数据的第一行是  m  n  两个整数（3  <   m,  n  <   100），表示正方形格子的行数和列数。 
接下来是  m  行数据。 
每行数据为  n  个被空格分开的数字。0  表示白格，1  表示黑格。 

接下来是一行数据：x  y  s  k,  其中x  y为整数，表示蚂蚁所在行号和列号（行号从上到下增长，列号从左到右增长，都是从0开始编号）。s  是一个大写字母，表示蚂蚁头的朝向，我们约定：上下左右分别用：UDLR表示。k  表示蚂蚁走的步数。

输出：

输出数据为一个空格分开的整数  p  q,  分别表示蚂蚁在k步后，所处格子的行号和列号。

样例输入：

5 6 
0 0 0 0 0 0 
0 0 0 0 0 0 
0 0 1 0 0 0 
0 0 0 0 0 0 
0 0 0 0 0 0 
2 3 L 5

样例输出：

1 3

dfs框架

 

上代码：

m, n = map(int, input().split()) #行数，列数
s = [] #储存输入的列表（表格）
for i in range(m): #输入二维列表
    s.append(list(map(int, input().split())))
a = list(input().split()) #储存x, y, d, step,也就是蚂蚁所在的(x,y)和朝向d，和目标要走的步数step
x = int(a[0])
y = int(a[1])
d = str(a[2])
step = int(a[3])
flag = 0 #避免回溯

def dfs(i, j, d, num):
    global flag
    if flag == 1: #如果flag=1就return
        return
    if num == step: #如果走的步数num达到了我们输入的step
        flag = 1 #flag就=1，也就是取上面return
        print(i, j) #打印出return的i，j
        return  #最后了，直接返回值

#"若蚂蚁在黑格，右转90度，将该格改为白格，并向前移一格"
#如果是黑棋子，刚开始的朝向d要分类，分成上下左右=>UDLR
    if s[i][j] == 1 and d == "U":  #s[i][j]为1是黑棋子，为0是白棋子
        d = "R"
        s[i][j] = 0
        dfs(i, j+1, d, num+1)

    if s[i][j] == 1 and d == "D":
        d = "L"
        s[i][j] = 0
        dfs(i, j-1, d, num+1)

    if s[i][j] == 1 and d == "L":
        d = "U"
        s[i][j] = 0
        dfs(i-1, j, d, num+1)

    if s[i][j] == 1 and d == "R":
        d = "D"
        s[i][j] = 0
        dfs(i+1, j, d, num+1)

#"若蚂蚁在白格，左转90度，将该格改为黑格，并向前移一格"
#如果是白棋子，刚开始的朝向d也要分类，分成上下左右=>UDLR

    if s[i][j] == 0 and d == "U":
        d = "L"
        s[i][j] = 1
        dfs(i, j-1, d, num+1)

    if s[i][j] == 0 and d == "D":
        d = "R"
        s[i][j] = 1
        dfs(i, j+1, d, num+1)

    if s[i][j] == 0 and d == "L":
        d = "D"
        s[i][j] = 1
        dfs(i+1, j, d, num+1)

    if s[i][j] == 0 and d == "R":
        d = "U"
        s[i][j] = 1
        dfs(i-1, j, d, num+1)
        
dfs(x, y, d, 0) #从num=0开始走

如有错误和问题欢迎在评论区指出噢！