蓝桥杯2014年第五届真题-兰顿蚂蚁-题解-思路清晰-防越界(Python代码)

def input_():#输入函数
    m,n = map(int,input().split())
    ls = [list(map(int,input().split())) for _ in range(m)]#多行列表限制输入
    x,y,s,k = input().split()#split()函数里面不要有任何参数
    x,y,k = map(int,[x,y,k])#整数转换
    move(x,y,s,k,m,n,ls)
def direction(s,w):#方向函数
    a = "ULDR".index(s)
    b = [(-1,0),(0,-1),(1,0),(0,1)]
    a += 1 if w==0 else -1
    dx,dy = b[a%4]
    Nexts = "ULDR"[a%4]#下一个方向
    return dx,dy,Nexts
def Transboundary(dx,dy,x,y,m,n):#防越界函数
    if (x+dx <= m and x+dx >= 0) and (y+dx <= n and y+dy >= 0):
        x,y = x+dx,y+dy
    return x,y
def move(x,y,s,k,m,n,ls):#移动函数
    count = 0
    while count<k:
        if ls[x][y] == 0:
            ls[x][y] = 1
            dx,dy,s= direction(s,0)
            x,y = Transboundary(dx,dy,x,y,m,n)
        else: 
            ls[x][y] = 0
            dx,dy,s= direction(s,1)
            x,y = Transboundary(dx,dy,x,y,m,n)
        count += 1
    print(x,y)
input_()