【美团】棋子翻转

题目描述

在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,1,1,1],[0,0,1,0],[0,1,1,0],[0,0,1,0]]

class Flip {
public:
    vector<vector<int> > flipChess(vector<vector<int> > A, vector<vector<int> > f) {
        // write code here
        if(A.size()<1)
            return A;
        
        int row = A.size(), col = A[0].size();
        int dx[4] = {0, -1, 0, 1};
        int dy[4] ={-1, 0, 1, 0}; 
        for(auto item:f)
        {
            int x = item[0]-1, y = item[1]-1;
            for(int i = 0;i < 4;i++)
            {
                if(x+dx[i]>-1&&x+dx[i]<row && y+dy[i]>-1&&y+dy[i]< col)
                    A[x+dx[i]][y+dy[i]]= 1-A[x+dx[i]][y+dy[i]];               
            }
        }
        return A;
    }
};