棋子翻转

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

代码如下:

import java.util.*;

public class Flip {
    public int[][] flipChess(int[][] A, int[][] f) {
        // write code here
        int row = A.length;
        int col = A[0].length;

        for(int i = 0;iint pointRow = f[i][0]-1;
            int pointCol = f[i][1]-1;

            int[] rows = {pointRow-1,pointRow,pointRow+1,pointRow};
            int[] cols = {pointCol,pointCol+1,pointCol,pointCol-1};

            for(int j = 0;j<4;j++){
                if(rows[j]>=0 && rows[j]=0 && cols[j]< col)
                    A[rows[j]][cols[j]] = A[rows[j]][cols[j]] == 1?0:1;
            }
        }
        return A;
    }
}

你可能感兴趣的:(机试题)