棋子翻转

题目描述

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

#include<iostream>
#include<vector>
using namespace std;
class Flip {
public:
    vector<vector<int> > flipChess(vector<vector<int> > A, vector<vector<int> > f) {
        // write code here
        int r=A.size();    //计算行数
        int c=A[0].size(); //计算列数
        int r1 = f.size();    //计算行数
        int c1 = f[0].size(); //计算列数
        cout << r1 << endl;
        for (int i = 0; i < r1; i++)
        {
            int row = f[i][0];
            int col = f[i][1];
            cout << row << " " << col << endl;
            if (row - 1 >= 1)
            {
                A[row - 1 - 1][col - 1] = abs(A[row - 1 - 1][col - 1] - 1);
            }
            if (row + 1 <= r )
            {
                A[row + 1 - 1][col - 1] = abs(A[row + 1 - 1][col - 1] - 1);
            }
            if (col - 1 >= 1)
            {
                A[row - 1][col - 1 - 1] = abs(A[row - 1][col - 1 - 1] - 1);
            }
            if (col+1<=c)
            {
                A[row - 1][col + 1 - 1] = abs(A[row - 1][col + 1 - 1] - 1);
            }
        }
        return A;
    }
};
int main()
{
    vector<vector<int>> A{ { 0, 0, 1, 1 }, { 1, 0, 1, 0 }, { 0, 1, 1, 0 }, { 0, 0, 1, 0 } };
    vector<vector<int>> f{ { 2,2}, {3,3 }, {4,4 }};
    Flip F;
    vector<vector<int>> b = F.flipChess(A, f);
    for (int i = 0; i < 4; i++)
    {
        for (int j = 0; j < 4; j++)
        {
            cout << b[i][j] << " ";
        }
        cout << endl;
    }
    system("pause");
    return 0;
}