Thought Works面试题

无人机题目

在这里if __name__=="__main__":
    # N：行数  M：列数 
    N,M=list(map(int,input().split(" ")))
    res=[[0]*M for _ in range(N)]
    matrix=dict()
    # 奇数*奇数 
    if N%2!=0 and M%2!=0:
        flag=0
        for i in range(0,N,2):
            if flag==0:
                for j in range(0,M,2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=1
                continue 
            if flag==1:
                for j in range(M-2,-1,-2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=0
        for key,value in matrix.items():
            i,j=eval(key)
            if 0<=i<=N-1 and 0<=j<=M-1:
                res[i][j]=value 
        for i in range(N):
            print("".join(res[i]))
    # 偶数*偶数 
    if N%2==0 and M%2==0:
        flag=0
        for i in range(0,N,2):
            if flag==0 and i<N-2:
                for j in range(0,M,2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=1
                continue 
            if flag==1 and i<N-2:
                for j in range(M-1,-1,-2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=0
            if flag==0 and i==N-2:
                for j in range(0,M):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
            if flag==1 and i==N-2:
                for j in range(M-1,-1,-1):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
        for key,value in matrix.items():
            i,j=eval(key)
            if 0<=i<=N-1 and 0<=j<=M-1:
                res[i][j]=value 
        for i in range(N):
            print("".join(res[i]))
    # 奇数*偶数 
    if N%2!=0 and M%2==0:
        flag=0
        for i in range(0,N,2):
            if flag==0:
                for j in range(0,M,2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=1
                continue 
            if flag==1:
                for j in range(M-1,-1,-2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=0
        for key,value in matrix.items():
            i,j=eval(key)
            if 0<=i<=N-1 and 0<=j<=M-1:
                res[i][j]=value 
        for i in range(N):
            print("".join(res[i]))
    # 偶数*奇数 
    if N%2==0 and M%2!=0:
        flag=0
        for i in range(0,N,2):
            if flag==0 and i<N-2:
                for j in range(0,M,2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=1
                continue 
            if flag==1 and i<N-2:
                for j in range(M-2,-1,-2):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
                flag=0
            if flag==0 and i==N-2:
                for j in range(0,M):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
            if flag==1 and i==N-2:
                for j in range(M-1,-1,-1):
                    s=input()
                    matrix["(%d,%d)"%(i,j)]=s[0]
                    matrix["(%d,%d)"%(i-1,j)]=s[1]
                    matrix["(%d,%d)"%(i,j+1)]=s[2]
                    matrix["(%d,%d)"%(i+1,j)]=s[3]
                    matrix["(%d,%d)"%(i,j-1)]=s[4]
        for key,value in matrix.items():
            i,j=eval(key)
            if 0<=i<=N-1 and 0<=j<=M-1:
                res[i][j]=value 
        for i in range(N):
            print("".join(res[i]))
插入代码片