代码
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
// 把nums1和nums2所有的可能相加
// <两数组之和,有几种搭配>
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums1.length; i++) {
for (int j = 0; j < nums2.length; j++) {
int sum = nums1[i] + nums2[j];
Integer value = map.get(sum);
if (value == null) {
map.put(sum, 1);
} else {
map.put(sum, ++value);
}
}
}
int ans = 0;
for (int i = 0; i < nums3.length; i++) {
for (int j = 0; j < nums4.length; j++) {
// 如果map中有值等于-(nums3[]+nums4[])就更新ans
int negSum = -(nums3[i] + nums4[j]);
Integer value = map.get(negSum);
if (value != null) {
ans += value;
}
}
}
return ans;
}
}
关键
注意map的方法怎么使用,比如get()、put()、contains()的传参和返回值
如果调用put()查找map中没有的Key,返回的是null