棋子反转

题目描述
在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]]

    vector<vector<int> > flipChess(vector<vector<int> > A, vector<vector<int> > f) {
        // write code here
        int size1 = A.size();
        int size2 = f.size();
        vector<vector<int>> res;
        if(size1 <= 0 || size2 <= 0)
            return res;
        for(int i = 0;i < size2;++i)
        {
            int f1 = f[i][0] -1;
            int f2 = f[i][1] -1;
            if((f1 - 1)>=0)
                A[f1-1][f2] = 1 - A[f1-1][f2] ;
            if((f1 + 1) < A.size())
                A[f1+1][f2]  = 1 - A[f1+1][f2];
            if((f2 -1)>=0)
                A[f1][f2-1] = 1- A[f1][f2-1];
            if((f2 + 1) < A[0].size())
                A[f1][f2+1]  = 1 - A[f1][f2+1];
        }
        return A;
    }
};

你可能感兴趣的:(校招编程题)