棋子翻转---- 美团2016研发工程师在线编程题

[编程题] 棋子翻转

在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> > vec, vector<vector<int> > f) {
// write code here
for ( int i = 0; i < f.size(); ++ i ) {
int x = f[i][0] - 1 ;
int y = f[i][1] - 1;

if ( x - 1 >= 0 ) {
int color = vec[x - 1][y] ;
if ( color == 0 ) color = 1 ;
else if ( color == 1 ) color = 0 ;
vec[x - 1][y] = color ;
}
if ( x + 1 <= 3 ) {
int color = vec[x + 1][y] ;
if ( color == 0 ) color = 1 ;
else if ( color == 1 ) color = 0 ;
vec[x + 1][y] = color ;
}

if ( y - 1 >= 0 ) {
int color = vec[x][y - 1] ;
if ( color == 0 ) color = 1 ;
else if ( color == 1 ) color = 0 ;
vec[x][y - 1] = color ;
}
if ( y + 1 <= 3 ) {
int color = vec[x][y + 1] ;
if ( color == 0 ) color = 1 ;
else if ( color == 1 ) color = 0 ;
vec[x][y + 1] = color ;
}
}
return vec ;
}
};