[蓝桥杯][2014年第五届真题]兰顿蚂蚁(Python实现)

**问题 1429: [蓝桥杯][2014年第五届真题] 兰顿蚂蚁

题目描述
兰顿蚂蚁，是于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

解题思路:

本题思路很简单，可以按着题目的要求把每一种情况都列出来。
1、确定方向：
可以先在纸上画出每种情况下的蚂蚁走向。如初始状态蚂蚁若为L ,即头向左，当处在黑格时，蚂蚁右转90度，即头的方向转为向上即U，若为白格时，蚂蚁左转90度，即头的方向变为D向下。
这样就可以定义出获取蚂蚁头部方向的函数。需要知道两个前提条件：一是蚂蚁的初始方向，二是当前方格是黑（1）还是白（0）

def getHead(i,d): #  i 为当前方格是黑还是白，黑为1，白为0 。d 为未转向时的原方向
    if i==1:
        if d=='L':d='U'
        elif d=='U':d='R'
        elif d=='R':d='D'
        elif d=='D':d='L'
    elif i==0:
        if d=='L':d='D'
        elif d=='U':d='L'
        elif d=='R':d='U'
        elif d=='D':d='R'
    return d

2、确定蚂蚁的下一步坐标
坐标表示说明：x表示某一行的第几个元素，即相当于列数，y表示第几行的某个元素，如x=2 , y=3 表示是第2行的第3个元素。
当转向后方向为上时，x的值为减小，向下时x的值为增加。如当前x=2,向上走时，x值加1变为3
同理向左 时y的值减1，向右时y的值加1

def getX(d,i,x):
    fx=getHead(i,d)
    if fx == 'D':return x + 1
    elif fx =='U': return x - 1
    else: return x
def getY(d,i,y):
    fx=getHead(i,d)
    if fx == 'R':return y + 1
    elif fx == 'L': return y - 1
    else: return y

3、确定输入数据的存储
Python中没有数组，对于一行输入进去的数据需要进行切分，再存入列表中

def listT(m,n):
    lt=[]
    for i in range(m):
        s=input().split()
        ll=[]
        for i in s:
            ll.append(int(i))
        lt.append(ll)
    return lt

4、主程序

m,n=map(int,input().split())  #先输入两个整数分别赋值给m,n
lt=listT(m,n)                # 调用输入数据的函数listT()
x,y,d,num=input().split()   #输入初始坐标，方向，及步数
x,y,num=int(x),int(y),int(num) #将输入数据转换为整型
#用步数控制循环次数  
for j in range(num):            
    xx=getX(d,lt[x][y],x)  #先获取到下一步的x、y坐标
    yy=getY(d,lt[x][y],y)
    d=getHead(lt[x][y],d)   # 再更新方向
    lt[x][y]=(lt[x][y]+1)%2  # 将走过的方格改变颜色，1为黑，0为白
    x=xx                     #对x、y的值进行更新
    y=yy
print('%d %d'%(x,y))        #以整型的数据输出x、y坐标，中间有空格