给你四个整数数组 nums1
、nums2
、nums3
和 nums4
,数组长度都是 n
,请你计算有多少个元组 (i, j, k, l)
能满足:nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0.
解题思路:
1.先遍历nums1和num2,求每两元素之和;
2.再遍历nums3和num4,求每两元素之和;
3.当作两数相加题目去解题。
public static int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int count=0;
int sum1; //a,b之和
int sum2; //与sum1相加为0的目标和
Map<Integer,Integer> map=new HashMap<Integer,Integer>();
//先遍历num1和num2,求a,b之和
for(int i:nums1){
for(int j:nums2){
sum1=i+j;
map.put(sum1,map.getOrDefault(sum1,0)+1);
}
}
//再遍历num3,num4,求c,d之和
for(int i:nums3){
for(int j:nums4){
sum2=0-(i+j);
if(map.containsKey(sum2)){
count=count+map.get(sum2);
}
}
}
return count;
}