算法训练 审美课 java 题解 194 （哈希）

问题描述

　　《审美的历程》课上有n位学生，帅老师展示了m幅画，其中有些是梵高的作品，另外的都出自五岁小朋友之手。老师请同学们分辨哪些画的作者是梵高，但是老师自己并没有答案，因为这些画看上去都像是小朋友画的……老师只想知道，有多少对同学给出的答案完全相反，这样他就可以用这个数据去揭穿披着皇帝新衣的抽象艺术了（支持帅老师^_^）。
　　答案完全相反是指对每一幅画的判断都相反。

输入格式

　　第一行两个数n和m，表示学生数和图画数；
　　接下来是一个n*m的01矩阵A：
　　如果aij=0，表示学生i觉得第j幅画是小朋友画的；
　　如果aij=1，表示学生i觉得第j幅画是梵高画的。

输出格式

　　输出一个数ans：表示有多少对同学的答案完全相反。

样例输入

3 2
1 0
0 1
1 0

样例输出

2

样例说明

　　同学1和同学2的答案完全相反；
　　同学2和同学3的答案完全相反；
　　所以答案是2。

数据规模和约定

　　对于50%的数据：n<=1000；
　　对于80%的数据：n<=10000；
　　对于100%的数据：n<=50000，m<=20。

---

解题思路:

通过将每个人所判断的“画”看做字符串，再转为数字，然后作为key存入map中，相同的数字次数作为value，遍历map时找正好成对的两个数字，各自出现的次数累加最终得到解。

java 代码：

import java.io.*;
import java.math.*;
import java.util.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String[] split = br.readLine().split(" ");
		int n = Integer.parseInt(split[0]);
		int m = Integer.parseInt(split[1]);
		
		StringBuilder s = new StringBuilder();
		for(int i = 0; i < m; i++) {
			s.append("1");
		}
		BigInteger flag = new BigInteger(s.toString());
		
		Map<BigInteger,Integer> map = new HashMap<>();
		for(int i = 0; i < n; i++) {
			StringBuilder str = new StringBuilder();
			split = br.readLine().split(" ");
			for(int j = 0; j < m; j++) {
				str.append(split[j]);
			}
			BigInteger bigInteger = new BigInteger(str.toString());
			Integer value = map.get(bigInteger);
			if(value == null) {
				map.put(bigInteger, 1);
			}else {
				map.put(bigInteger, value + 1);
			}
		}
		
		int ans = 0;
		Iterator<BigInteger> iterator = map.keySet().iterator();
		while(iterator.hasNext()) {
			BigInteger key = iterator.next();
			Integer value = map.get(key);
			BigInteger other = flag.subtract(key);
			Integer cot = map.get(other);
			if(cot != null) {
				ans += value * cot;
				map.put(key, null);
			}
		}
		System.out.println(ans);
	}
}

提交截图：