【蓝桥杯】兰顿蚂蚁

 　　兰顿蚂蚁，是于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

---

一、分步解析

1、输入代码

m,n = map(int,input().split()) #map将m、n以int形式存放
arr = [0] *m  # 准备好每一行的数组
for i in range(m):
    arr[i] = [int(i) for i in input().split()] #将一行的数字填入到arr中，最后形成长方形
x,y,s,k = input().split()  #输入，type为str
x,y,k = int(x),int(y),int(k)  #转化为int
step = 0  #刚开始步数设为0

2、用一个函数将URDL转换成1234，即接下来的代码+1为右转，-1为左转

def head(s):
    if s == 'U':
        return 1
    if s == 'R':
        return 2
    if s == 'D':
        return 3
    if s == 'L':
        return 4
    return 0

3、转向变色前进，循环结束后输出k步后的坐标

s = head(s)  #获取初始数字的方向
while step < k:  
    if arr[x][y] == 1:  #如果为黑格
        s += 1   #加1右转
        if s == 5:  #当为4是右转应该变为1
            s = 1
        arr[x][y] = 0  #转完后格子变白
    else:
        if s == 0:  
            s = 4   #限制不能为0，0时相当转了一周再回到4
        s -= 1  #白格子左转-1
        arr[x][y] = 1  #格子变黑
    if s == 1:  #转向变色后前进一步，前进的x,y取决于s的方向
        x -= 1
    elif s == 2:
        y += 1
    elif s == 3:
        x += 1
    else:
        y -= 1
    step += 1
print(x,y)

二、正确代码

1、初始版（自己一开始写的

m,n = map(int,input().split())
arr = [0] *m 
for i in range(m):
    arr[i] = [int(i) for i in input().split()]
x,y,s,k = input().split()
x,y,k = int(x),int(y),int(k)
step = 0
def headDirection(s,color): #判断头转动后的方向
    if s == 'U'and color == 0 or s == 'D' and color == 1:  
        s = 'L'
    elif s == 'L'and color == 0 or color == 1 and s == 'R':
        s = 'D'
    elif s == 'D'and color == 0 or color == 1 and s == 'U':
        s = 'R'
    else:
        s = 'U'
    return s
def nextStep(s,x,y): #判断方向后前进一步
    if s == 'U':
        return x - 1,y
    elif s == 'L':
        return x,y - 1
    elif s == 'R':
        return x,y + 1
    else:
        return x + 1,y
        
while step < k: 
    s = headDirection(s,arr[x][y])  #调用转动后头方向的函数
    if arr[x][y] == 0: #若为白色则更改为黑色
        arr[x][y] = 1
    else:
        arr[x][y] = 0
    step += 1  #步数加1
    x,y = nextStep(s,x,y) # 调用前进一步的函数
print(x,y)

2、升级版（上面分析的代码

m,n = map(int,input().split())
arr = [0] *m 
for i in range(m):
    arr[i] = [int(i) for i in input().split()]
x,y,s,k = input().split()
x,y,k = int(x),int(y),int(k)
step = 0
def head(s):
    if s == 'U':
        return 1
    if s == 'R':
        return 2
    if s == 'D':
        return 3
    if s == 'L':
        return 4
    return 0
s = head(s)
while step < k:
    if arr[x][y] == 1:
        s += 1
        if s == 5:
            s = 1
        arr[x][y] = 0
    else:
        if s == 0:
            s = 4
        s -= 1
        arr[x][y] = 1
    if s == 1:
        x -= 1
    elif s == 2:
        y += 1
    elif s == 3:
        x += 1
    else:
        y -= 1
    step += 1
print(x,y)