算法训练 审美课 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。

解题思路：

由于数据规模太大，如果暴力复杂度为 50000*50000*20 ，肯定会超时。可以将每个人的答案都拼接成一个字符串，完全相反即为两个字符串的数值和全为1，否则即不满足题意。但这样只能把复杂度降到 50000*50000 ，仍然很大。这里可以通过利用哈希表来实现空间换取时间。由于m可以达到20，所以不可能开一个10的20次的数组，只能通过map配BigInteger实现哈希表存储。直接将时间复杂度降到O(n)，空间复杂度为O(n）。在遍历map时，要将一对完全相反的key的value做乘积，还要将此时的value置为null，以避免重复计算。

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 builder2 = new StringBuilder();
		for(int i = 0; i < m;i++) {
			builder2.append("1");
		}
		BigInteger flag = new BigInteger(builder2.toString());
		
		Map<BigInteger,Integer> map = new HashMap<>();
		for(int i = 0; i < n;i++) {
			StringBuilder builder = new StringBuilder();
			split = br.readLine().split(" ");
			for(int j = 0; j < m;j++) {
				builder.append(split[j]);
			}
			BigInteger num = new BigInteger(builder.toString());
			Integer value = map.get(num);
			if(value == null) {
				map.put(num, 1);
			}else {
				map.put(num, value + 1);
			}
		}
		Iterator<BigInteger> iterator = map.keySet().iterator();
		int ans = 0;
		while(iterator.hasNext()) {
			BigInteger key = iterator.next();
			BigInteger tag = flag.subtract(key);
			Integer value = map.get(tag);
			Integer value1 = map.get(key);
			if(value != null) {
				ans += value * value1;//注意是两数出现次数的乘积
				map.put(key, null);//避免重复计算
			}
		}
		System.out.println(ans);
	}
}

提交截图：