算法,不如说它是一种思考方式
算法专栏: 123
nums1
、nums2
、nums3
和 nums4
,数组长度都是 n
,请你计算有多少个元组 (i, j, k, l)
能满足:0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-2^28 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 2^28
4个数组如果使用暴力解法就需要4层for
循环,那时间复杂度将是O(n4)。
所以和上一题LeetCode:1. 两数之和类似,本题还是可以使用哈希表。我们可以把四个数组分为两个组,分别对每个组求和(时间复杂度O(n2) ),将第一组求和结果加入到map
的key
中,value
就是出现的次数。第二组就用于寻找map
中是否有相反数,最后统计次数返回。
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map<Integer,Integer> map=new HashMap<>();
for (int i = 0; i < nums1.length; i++) {
for (int j = 0; j < nums2.length; j++) {
if(!map.containsKey(nums1[i]+nums2[j])){
map.put(nums1[i]+nums2[j],1);
}else{
map.put(nums1[i]+nums2[j],map.get(nums1[i]+nums2[j])+1 );
}
}
}
int ans=0;
for (int i = 0; i < nums3.length; i++) {
for (int j = 0; j < nums4.length; j++) {
if(map.containsKey(-(nums3[i]+nums4[j]))){
ans+=map.get(-(nums3[i]+nums4[j]));
}
}
}
return ans;
}
}
返回第一页。☝
☕物有本末,事有终始,知所先后。
☝☝☝☝☝我的CSDN☝☝☝☝☝☝