蓝桥杯-兰顿蚂蚁

'''
兰顿蚂蚁，是于 1986 年，由克里斯·兰顿提出来的，属于细胞自动机的一种。


平面上的正方形格子被填上黑色或白色。在其中一格正方形内有一只“蚂蚁”。

蚂蚁的头部朝向为：上下左右其中一方。

蚂蚁的移动规则十分简单：
若蚂蚁在黑格，右转 90 度，将该格改为白格，并向前移一格；
若蚂蚁在白格，左转 90 度，将该格改为黑格，并向前移一格。

规则虽然简单，蚂蚁的行为却十分复杂。刚刚开始时留下的路线都会有接近对称，像是
会重复，但不论起始状态如何，蚂蚁经过漫长的混乱活动后，会开辟出一条规则的“高速公
路”。


蚂蚁的路线是很难事先预测的。
你的任务是根据初始状态，用计算机模拟兰顿蚂蚁在第 n 步行走后所处的位置。
输入格式


输入数据的第一行是 m n 两个整数（3 < m, n < 100 ），表示正方形格子的行数和列
数。
接下来是 m 行数据。

每行数据为 n 个被空格分开的数字。0 表示白格，1 表示黑格。


输入：
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      注（x,y,朝向，步数）
样例输出
1 3          注（x,y）


'''

'''
 先定义左转 和 右转

'''
# 定义右转
def turn_right(s):
    '''
    上下左右分别用：UDLR 表示
    :param s:
    :return:
    '''
    if(s == 'U'):
        return 'R'
    if(s == 'R'):
        return "D"
    if(s == "D"):
        return "L"
    if( s == 'L' ):
        return "U"


# 定义右转
def turn_left(s):
    '''
    上下左右分别用：UDLR 表示
    :param s:
    :return:
    '''
    if(s == 'U'):
        return 'L'
    if(s == 'L'):
        return "D"
    if(s == "D"):
        return "R"
    if( s == 'R' ):
        return "U"


'''
定义前进
'''

def my_forward(s,x,y):
    if( s == "U"):
        return x,y-1
    if (s == "D"):
        return x, y + 1
    if (s == "L"):
        return x - 1, y
    if (s == "R"):
        return x + 1, y



'''
开始前进
'''

mat = [
    [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]
]
g_x = 2
g_y = 3
g_s = 'L'
g_k = 5


def operate_one_step(mat):
    '''
    若蚂蚁在黑格，右转 90 度，将该格改为白格，并向前移一格；
    若蚂蚁在白格，左转 90 度，将该格改为黑格，并向前移一格。
    :param mat:
    :param x:
    :param y:
    :param s:
    :return:
    '''
    global g_x
    global g_y
    global g_s
    global g_k

    # 黑色
    if( mat[g_x][g_y] == 1):
        # 若蚂蚁在黑格，右转 90 度，将该格改为白格，并向前移一格；
        g_s = turn_right(g_s)
        g_x, g_y = my_forward(g_s , g_x , g_y)

    # 白色
    if (mat[g_x][g_y] == 0):
        # 若蚂蚁在黑格，右转 90 度，将该格改为白格，并向前移一格；

        g_s = turn_left(g_s)


        g_x, g_y = my_forward(g_s, g_x, g_y)

operate_one_step(mat)
print(g_x,g_y,g_s)

operate_one_step(mat)
print(g_x,g_y,g_s)

operate_one_step(mat)
print(g_x,g_y,g_s)

operate_one_step(mat)
print(g_x,g_y,g_s)

operate_one_step(mat)
print(g_x,g_y,g_s)


print("final")
print(g_x-1 , g_y-1 , g_s)
# for each in mat:
#     print(each)


# print(g_x,g_y)