1.题目描述:
给你四个整数数组nums1、nums2、nums3和nums4,数组长度都是n,请你计算有多少个元组 (i, j, k, l) 能满足:0 <= i, j, k, l < n;nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0。
2.使用hashmap:
nums1和nums2一组双层for循环遍历将和存入hashmap中,nums3和nums4一组双层for循环遍历,这种算法中包含重复项,也可以三一分组。
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map<Integer, Integer> map = new HashMap<>();
int res = 0;
for (int i = 0; i < nums1.length; i++) {
for (int j = 0; j < nums2.length; j++) {
int val = nums1[i] + nums2[j];
if (map.containsKey(val)) map.put(val, map.get(val) + 1);
else map.put(val, 1);
}
}
for (int i = 0; i < nums3.length; i++) {
for (int j = 0; j < nums4.length; j++) {
int val = nums3[i] + nums4[j];
if (map.containsKey(-val)) res += map.get(-val);
}
}
return res;
}
}