算法-每天一道题（5）-棋子翻转

在4×4的棋盘上摆满了黑白棋子，黑白两色的位置和数目随机其中左上角坐标为(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,1,1,1],[0,0,1,0],[0,1,1,0],[0,0,1,0]]

Python

# -*- coding:utf-8 -*-
 
class Flip:
    def flipChess(self, A, f):
        # write code here
        for site in f:
            x = site[0]-1
            y = site[1]-1
            if x > 0 and x < 3:
                A[x-1][y] = int(not A[x-1][y])
                A[x+1][y] = int(not A[x+1][y])
            elif x == 0:
                A[x+1][y] = int(not A[x+1][y])
            else:
                A[x-1][y] = int(not A[x-1][y])
             
            if y > 0 and y < 3:
                A[x][y-1] = int(not A[x][y-1])
                A[x][y+1] = int(not A[x][y+1])
            elif y == 0:
                A[x][y+1] = int(not A[x][y+1])
            else:
                A[x][y-1] = int(not A[x][y-1])
        return A
'''
解题思路：
这道题并不难，只需要把中心点的位置判断一下，
然后将周围的点翻转便可以了
注意的一点是，这里的输入是从0开始的矩阵，
以及从1开始的位置坐标，所以位置要-1
'''