【美團】（數據方向-筆試）棋子翻轉【Python】

題目描述

在4x4的棋盤上擺滿了黑白棋子，黑白兩色的位置和數目隨機其中左上角座標爲(1,1),右下角座標爲(4,4)
現在依次有一些翻轉操作，要對一些給定支點座標爲中心的上下左右四個棋子的顏色進行翻轉，請計算出翻轉後的棋盤顏色。

給定兩個數組A和f,分別爲初始棋盤和翻轉位置。其中翻轉位置共有3個。請返回翻轉後的棋盤。

輸入樣例

[[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]],[[2,2],[3,3],[4,4]]

輸出樣例

[[0,0,1,1],[1,0,1,0],[0,1,1,0],[0,0,1,0]],[[2,2],[3,3],[4,4]]

棋子翻轉（Python）

def flip(A, f):
    def getNum(i):
        return 0 if i == 1 else 1

    for i in f:
        row, col = i[0] - 1, i[1] - 1
        if row > 0:
            A[row - 1][col] = getNum(A[row - 1][col])
        if row < 3:
            A[row + 1][col] = getNum(A[row + 1][col])
        if col > 0:
            A[row][col - 1] = getNum(A[row][col - 1])
        if col < 3:
            A[row][col + 1] = getNum(A[row][col + 1])
    return A

A = eval(input())
f = eval(input())
B = flip(A,f)
print (B)