美团-棋子翻转-Java

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

import org.junit.Test;

/**
 * 题目描述 在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]]
 * 
 * @author 崔洪振367
 * @version 创建时间：2017年7月7日 下午1:01:55
 */
public class 棋子翻转 {
	/**
	 * 解题思路：
	 * 1、声明一个HashMap，<String , Integer>
	 * 2、遍历翻转数组，求出每个节点对应的四个节点位置，以字符串形式保存到hashmap键中，出现次数保存到值中 
	 * 3、遍历HashMap，修改原棋盘
	 */
	HashMap<String, Integer> hashMap = null;
	
	@Test
	public void test(){
		int[][] A = {{0,0,1,1},{1,0,1,0},{0,1,1,0},{0,0,1,0}};
		int[][] f = {{2,2},{3,3},{4,4}};
		int[][] result = flipChess(A, f);
		for(int i=0; i<result.length; i++){
			for(int j=0; j<result[0].length; j++){
				System.out.print(result[i][j] + ",");
			}
			System.out.println();
		}
	}

	public int[][] flipChess(int[][] A, int[][] f) {
		hashMap = new HashMap<>();

		int aLen = A.length;
		int aaLen = A[0].length;

		// 遍历翻转的矩阵，找出四个点的坐标
		for (int i = 0; i < f.length; i++) {// 题目中f.length的大小为3
			int point0 = f[i][0];
			int point1 = f[i][1];
			String hmKey = "";
			//上
			if ((point0 - 1) > 0 && (point0 - 1) <= aLen && point1 > 0 && point1 <= aaLen) {
				hmKey = (point0-1)+","+point1;
				addElementToHashMap(hashMap, hmKey);
			}
			//下
			if ((point0+1) > 0 && (point0+1) <= aLen && point1 > 0 && point1 <= aaLen) {
				hmKey = (point0+1)+","+point1;
				addElementToHashMap(hashMap, hmKey);
			}
			//左
			if (point0 > 0 && point0 <= aLen && (point1-1) > 0 && (point1-1) <= aaLen) {
				hmKey = point0+","+(point1-1);
				addElementToHashMap(hashMap, hmKey);
			}
			//右
			if (point0 > 0 && point0 <= aLen && (point1+1) > 0 && (point1+1) <= aaLen) {
				hmKey = point0+","+(point1+1);
				addElementToHashMap(hashMap, hmKey);
			}
		}
		
		//遍历HashMap，修改原棋盘
		Iterator<Entry<String, Integer>> iterator =  hashMap.entrySet().iterator();
		while(iterator.hasNext()){
			Map.Entry<String, Integer> entry = (Entry<String, Integer>) iterator.next();
			int v = entry.getValue();
			String k = entry.getKey();
			//System.out.println("key:"+k+","+"value:"+v);
			if(v%2!=0){
				String[] p = k.split(",");
				int x = Integer.parseInt(p[0]);
				int y = Integer.parseInt(p[1]);
				A[x-1][y-1] = A[x-1][y-1]==0?1:0;
			}
		}
		return A;
	}

	public void addElementToHashMap(HashMap<String, Integer> hp, String fz) {
		if (hashMap.containsKey(fz)) {
			int value = hashMap.get(fz);
			hashMap.put(fz, value+1);
		} else {
			hashMap.put(fz, 1);
		}
	}

}