Python 蓝桥杯 兰顿蚂蚁

一、题目描述

兰顿蚂蚁，是于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
样例输入
3 3
0 0 0
1 1 1
1 1 1
1 1 U 6
样例输出
0 0

二、解题思路

重点在于黑白格和蚂蚁头方向。可以首先使用if语句判断蚂蚁此时所在的位置是黑格还是白格。如果是是黑格，向右转90度，相当于一个列表上右下左进行循环，此时如果方向是左需要特殊处理。同理的，如果是白格，相当于是上左下右进行循环，和黑格的循环方向相反。结合深度遍历可以设置一个列表代表方向，与代表上下左右的列表进行对应，如下：

sign=['U','R','D','L']
move=[[-1,0],[0,+1],[+1,0],[0,-1]]

再令一个参数a表示此时蚂蚁头的方向，即sign列表的下标即可。

三、代码

m,n=list(map(int,input().split()))
l =[]
sign=['U','R','D','L']
move=[[-1,0],[0,+1],[+1,0],[0,-1]]
for i in range(m):
    l.append(list(map(int,input().split())))
M = list(input().split())
X,Y,s,k = int(M[0]),int(M[1]),M[2],int(M[3])

def dfs(x,y,a):
    global X,Y,k
    if l[x][y]==1:# 黑格
        l[x][y]=0
        if a!=3:
            a+=1
            x+=move[a][0]
            y+=move[a][1]
        else:
            a=0
            x+=move[0][0]
            y+=move[0][1]
    else:
        l[x][y]=1
        if a!=0:
            a-=1
            x+=move[a][0]
            y+=move[a][1]
        else:
            a=3
            x+=move[3][0]
            y+=move[3][1]
    if 0<=x<=m and 0<=y<=n and k>0:
        k-=1
        if k==0:
            print(x,y)
            return 0
        dfs(x,y,a)
        
    
dfs(X,Y,''.join(sign).find(s))

注意，先判断k是否为0.再进行递归。