在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]]
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)