题目链接
思路:分组+哈希表
分析:四个树状A、B、C、D,将AB分为一组,CD分为一组。
首先计算AB一共可以加出多少个不同的数字来,并且记录每个数字有多少种情况可以加出来。(记录在哈希表种)
然后再计算0减去CD的值相加,并且去哈希表中找,看有几种情况。
也就是将A+B记在哈希表中,然后在哈希表中寻找-(C+D),看有多少种情况,然后进行累加即可得到答案。
代码:
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int len = nums1.length;
//key是记录A+B,val是记录多少用不同的情况
HashMap<Integer, Integer> map = new HashMap<>();
int res=0;
for(int i=0;i<len;i++){
for(int j=0;j<len;j++){
//将情况+1
map.put(nums1[i]+nums2[j],map.getOrDefault(nums1[i]+nums2[j],0)+1);
}
}
for(int i=0;i<len;i++){
for(int j=0;j<len;j++){
//-(C+D)在哈希表中寻找。并且累加答案
res = res + map.getOrDefault(-nums3[i]-nums4[j],0);
}
}
return res;
}
}