校招真题 美团 棋子翻转

校招真题 美团 棋子翻转

题目

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

---

考查点

数组、模拟、编程基础

---

思路

边界判断
0 和 1 转换的小技巧，1-x

---

程序

import java.util.*;

public class Flip {
    public int[][] flipChess(int[][] A, int[][] f) {
        // write code here
        for (int i = 0; i < f.length; i ++){
            int x = f[i][0]-1;
            int y = f[i][1]-1;

            if (x >= 0 && x < A.length){
                if (y-1 >= 0 && y-1 < A.length) // 上
                    A[x][y-1] = 1 - A[x][y-1];
                if (y+1 >= 0 && y+1 < A.length) // 下
                    A[x][y+1] = 1 - A[x][y+1];
            }

            if (y >= 0 && y < A.length){
                if (x-1 >= 0 && x-1 < A.length) // 右
                    A[x-1][y] = 1 - A[x-1][y];
                if (x+1 >= 0 && x+1 < A.length) // 左
                    A[x+1][y] = 1 - A[x+1][y];
            }
        }
        return A;

    }
}