454. 四数之和 II
给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ,使得 A[i] + B[j] + C[k] + D[l] = 0。
为了使问题简单化,所有的 A, B, C, D 具有相同的长度 N,且 0 ≤ N ≤ 500 。所有整数的范围在 -228 到 228 - 1 之间,最终结果不会超过 231 - 1 。
例如:
输入:
A = [ 1, 2]
B = [-2,-1]
C = [-1, 2]
D = [ 0, 2]
输出:
2
解释:
两个元组如下:
- (0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
- (1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0
本题解题思路:
1.首先定义一个map,key用来存储 a+b的和,value放a+b值出现的次数;
2.遍历数组A,B,统计a+b的所有结果及次数,存储在map中;
3.定义sum,用来统计A[i] + B[j] + C[k] + D[l] = 0出现的次数;
4.遍历数组C,D,如果在map找到0-(c+d),那么用sum将map中key对应的value统计出来(也就是key出现的次数);
5.最后返回sum;
科普一个HashMap中的一个方法:
map.getOrDefault(Object key , V defaultValue)
这个方法的作用是如果key对应的value上没有值,那么则value为一个你自己设置的一个默认值,例如本题,我们将默认值设置为0
下图为此方法的源代码,有助于大家理解
default V getOrDefault(Object key, V defaultValue) {
V v;
return (((v = get(key)) != null) || containsKey(key))
? v
: defaultValue;
}
我们拿本题举个例子:
我们第一次遍历数组A,B完成后map中是这样的
key = 1±2 , value = 1
第二次
key = 1+ (-1) , value = 1
第三次
key = 2 + (-2), value = 2;
第四次…
本题java代码,其中有注释
class Solution {
public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
//key:a+b的值,value:a+b出现的次数
Map<Integer,Integer> map = new HashMap<Integer,Integer>();
for (int u : A) {
for (int v : B) {
//把相应的值放入map中
map.put(u + v , map.getOrDefault(u + v , 0) + 1);
}
}
int sum = 0;
for (int u : C) {
for (int v : D) {
//相当于0-(c+d)= a+b
if (map.containsKey(-u - v)) {
sum += map.get(-u - v);
}
}
}
return sum;
}
}```
若有误,望指教!