棋子翻转（牛客网）

题目描述

在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) {
        int[][] res = A;

		HashSet<Integer[]> help = new HashSet<>();
		for (int i = 0; i < f.length; i++) {
			// hang
			if (f[i][0] == 1) {
				Integer[] temp = new Integer[] { f[i][0] + 1, f[i][1] };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
			} else if (f[i][0] == 4) {
				Integer[] temp = new Integer[] { f[i][0] - 1, f[i][1] };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
			} else {
				Integer[] temp = new Integer[] { f[i][0] + 1, f[i][1] };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
				temp = new Integer[] { f[i][0] - 1, f[i][1] };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
			}

			// lie
			if (f[i][1] == 1) {
				Integer[] temp = new Integer[] { f[i][0], f[i][1] + 1 };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
			} else if (f[i][1] == 4) {
				Integer[] temp = new Integer[] { f[i][0], f[i][1] - 1 };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
			} else {
				Integer[] temp = new Integer[] { f[i][0], f[i][1] + 1 };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
				temp = new Integer[] { f[i][0], f[i][1] - 1 };
				if (help.contains(temp)) {
					help.remove(temp);
				} else {
					help.add(temp);
				}
			}
		}

		Iterator<Integer[]> iterator = help.iterator();

		while (iterator.hasNext()) {
			Integer[] tmp = iterator.next();
			
			if (res[tmp[0] - 1][tmp[1] - 1] == 0) {
				res[tmp[0] - 1][tmp[1] - 1] = 1;
			} else {
				res[tmp[0] - 1][tmp[1] - 1] = 0;
			}
		}

		return res;
    }
}

为什么用了很多代码来处理f 数组，不想对A数组进行重复的改动和检测，先将要翻转的位置处理好。