棋子翻轉

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

代碼:

public class Flip {

    public int[][] flipChess(int[][] A, int[][] f) {
        // write code here
        int m = A.length;
        int n = A[0].length;
        for ( int i = 0 ; i < f.length ; i++){
            int row = f[i][0] - 1;
            int col = f[i][1] - 1;
            if ( row -1 >= 0 ){
                A[row-1][col] = (A[row-1][col] == 0)? 1:0;
            }
            if ( row + 1 < n ){
                A[row+1][col] = (A[row+1][col] == 0)? 1:0;
            }
            if ( col -1 >= 0 ){
                A[row][col-1] = (A[row][col-1] == 0)? 1:0;
            }
            if ( col + 1 < m ){
                A[row][col+1] = (A[row][col+1] == 0)? 1:0;
            }
        }
        return A;
    }
}           

複制