解析:(分组+HashMap)
题目要求:四个整数数组 nums1
、nums2
、nums3
和 nums4
,计算有多少组 (i, j, k, l)
能满足:nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
大体思路:
可以使用一个哈希表,key存放nums1和nums2所有元素组合情况的和,value存放这个和出现了多少次。再去哈希表中依次查找0-(nums3和nums4所有元素组合情况的和),如果找到,结果加上这个值在哈希表中的value,也就是出现了多少次。
注意点:使用哈希表时间复杂度更低
使用哈希表的时间复杂度是O(2n平方),因为是需要使用两个两层的for循环,而哈希表查找的时间复杂度为O(1).
使用暴力法需要4层的for循环,时间复杂度是O(n的4次方)。
代码:
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
HashMap<Integer, Integer> map = new HashMap<>();
for (int i : nums1) {
for (int j : nums2) {
int temp = i + j;
if (map.containsKey(temp)) {
map.put(temp, map.get(temp) + 1);
}
else {
map.put(temp, 1);
}
}
}
int res = 0;
for (int i : nums3) {
for (int j : nums4) {
int temp = i + j;
if (map.containsKey(0-temp)) {
res += map.get(0-temp);
}
}
}
return res;
}
}