454. 四数相加 II - 力扣(LeetCode) (leetcode-cn.com)https://leetcode-cn.com/problems/4sum-ii/先遍历前两个数组元素之间的和,然后记录到Map中(key是和,value是和出现的次数)。
然后遍历后两个数组元素之间的和,判断这个和的相反数书否在Map中,若在,count 就加上此相反数出现的次数。
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map<Integer,Integer> map = new HashMap<>();
//先计算前两个数组的和,并统计每个和出现的次数
for(int i:nums1){
for(int j:nums2){
if (map.containsKey(i+j))
map.put(i+j,map.get(i+j)+1);
else
map.put(i+j,1);
}
}
//计算后两个数组的和,count+=和的相反数出现的次数
int count = 0;
for(int i:nums3){
for(int j:nums4){
if(map.containsKey(-(i + j))){
count+=map.get(-(i+j));
}
}
}
return count;
}
}