思路
相当于A+B+C+D=0,求ABCD多少种可能性,每个数组数据量10^2,刚开始的时候我是遍历A,B,C,然后看-A-B-C在不在D对应的Map中,然后将他们元素对应的次数相乘即为答案,时间超限,所以,考虑只遍历AB和CD,然后看他们的和是否为相反数。
题目描述
给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
代码
package Hash;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
public class _454四数相加 {
public static int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
HashMap<Integer,Integer> map1=new HashMap<Integer,Integer>();
for(int i=0;i<nums1.length;i++) {
map1.put(nums1[i], map1.getOrDefault(nums1[i], 0)+1);
}
HashMap<Integer,Integer> map2=new HashMap<Integer,Integer>();
for(int i=0;i<nums2.length;i++) {
map2.put(nums2[i], map2.getOrDefault(nums2[i], 0)+1);
}
HashMap<Integer,Integer> summap12=new HashMap<Integer,Integer>();
for(Map.Entry<Integer, Integer> map11: map1.entrySet()) {
for(Map.Entry<Integer, Integer> map22:map2.entrySet()) {
summap12.put(map11.getKey()+map22.getKey(), summap12.getOrDefault(map11.getKey()+map22.getKey(), 0)+map11.getValue()*map22.getValue());
}
}
HashMap<Integer,Integer> map3=new HashMap<Integer,Integer>();
for(int i=0;i<nums3.length;i++) {
map3.put(nums3[i], map3.getOrDefault(nums3[i], 0)+1);
}
HashMap<Integer,Integer> map4=new HashMap<Integer,Integer>();
for(int i=0;i<nums4.length;i++) {
map4.put(nums4[i], map4.getOrDefault(nums4[i], 0)+1);
}
HashMap<Integer,Integer> summap34=new HashMap<Integer,Integer>();
for(Map.Entry<Integer, Integer> map33: map3.entrySet()) {
for(Map.Entry<Integer, Integer> map44:map4.entrySet()) {
summap34.put(map33.getKey()+map44.getKey(), summap34.getOrDefault(map33.getKey()+map44.getKey(), 0)+map33.getValue()*map44.getValue());
}
}
/*for(Map.Entry<Integer, Integer> map12:summap12.entrySet()) {
System.out.println(map12.getKey()+" "+map12.getValue());
}
System.out.println("********");
for(Map.Entry<Integer, Integer> map34:summap34.entrySet()) {
System.out.println(map34.getKey()+" "+map34.getValue());
}*/
int count=0;
for(Map.Entry<Integer, Integer> map12:summap12.entrySet()) {
if(summap34.containsKey(-map12.getKey())) {
count=count+map12.getValue()*summap34.get(-map12.getKey());
}
}
//System.out.println(count);
return count;
}
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
while(sc.hasNext()) {
int n1=sc.nextInt();
int nums1[]=new int[n1];
for(int i=0;i<n1;i++) {
nums1[i]=sc.nextInt();
}
int n2=sc.nextInt();
int nums2[]=new int[n2];
for(int i=0;i<n2;i++) {
nums2[i]=sc.nextInt();
}
int n3=sc.nextInt();
int nums3[]=new int[n3];
for(int i=0;i<n3;i++) {
nums3[i]=sc.nextInt();
}
int n4=sc.nextInt();
int nums4[]=new int[n4];
for(int i=0;i<n4;i++) {
nums4[i]=sc.nextInt();
}
fourSumCount(nums1,nums2,nums3,nums4);
}
}
}
总结
Map是一种快速查询的利器!!!