![在这里插入图片描述](https://img-blog.csdnimg.cn/direct/fa5ba52599664606864afbf705ba39d2.png)
- 共4个数组A,B,C,D,如果暴力枚举,就需要4个for循环,枚举每一种组合。则时间复杂度O(
n
4
n^4
n4)
- 但是我们现在分为两组,A和B的每一种组合,和C和D的每一种组合. 时间复杂度就变成了2个O(
n
2
n^2
n2)
关键:
假设AB组合,有一个值是1.CD组合有一个值是-1.那么这个整体的组合,相加为0,显然是题目想要我们找到的。- 问题时,我们如何将其对应到一起。我们发现
这个完全符合题意的组合。如果CD组合的值取负,就和AB组合的值一样了
。例如AB组合的数是-1000,CD组合的数是1000,那么CD取负变-1000. 此时两个组合的数就一样了,都是-1000. - 所以我们可以建立
n
2
n^2
n2大小的hash表,存储所有AB组合的数,然后只需要判断CD组合取负后,是否和hash表中元素对应,如果对应,那么它们相加就一定归0.
而对于hash表来说,我们可以用做题的老套路,用数组充当hash表,做题效率更快,但是工作中可不能用这玩意。同时我也会给出用实实在在的Hash表来存储的方法,它的效率略逊于数组直接存取,但是工作中肯定得用这个。而对于做题来说,虽然hash表存储比用数组慢了2ms。但是就是这2ms,却超过了65%的人。而实际工作中,没必要为了这2ms来承受使用数组的风险。
代码:使用Map集合:时间复杂度O(n^ 2 ) 空间复杂度O(n^ 2),使用数组:时间复杂度O(n^2) 空间复杂度O(value) 其中value是数组中元素的取值范围 |
---|
![在这里插入图片描述](https://img-blog.csdnimg.cn/direct/4d5e37e6d29648e3a3b7e1ca4722e26a.png)
下面代码中,一共两个解决方法的代码,注意区分。
class Solution {
public int[] getMaxMin(int[] nums) {
int min = nums[0], max = nums[0];
for (int num: nums) {
if (num < min) min = num;
else if (num > max) max = num;
}
return new int[] {max, min};
}
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int[] max_min_1 = getMaxMin(nums1);
int[] max_min_2 = getMaxMin(nums2);
int[] max_min_3 = getMaxMin(nums3);
int[] max_min_4 = getMaxMin(nums4);
int max = Math.max(max_min_1[0] + max_min_2[0], -(max_min_3[1] + max_min_4[1]));
int min = Math.min(max_min_1[1] + max_min_2[1], -(max_min_3[0] + max_min_4[0]));
int [] hash = new int [max - min + 1];
int ans = 0;
for (int num1: nums1) {
for (int num2: nums2) {
++hash[num1 + num2 - min];
}
}
for (int num3: nums3) {
for (int num4: nums4) {
ans += hash[-num3 - num4 - min];
}
}
return ans;
}
public int fourSumCount1(int[] A, int[] B, int[] C, int[] D) {
Map<Integer,Integer> countAB = new HashMap<Integer,Integer>();
for(int u : A)
for(int v : B)
countAB.put(u + v,countAB.getOrDefault(u + v , 0) + 1);
int ans = 0;
for(int u : C)
for(int v : D)
if(countAB.containsKey(-u - v))
ans += countAB.get(-u - v);
return ans;
}
}