原题链接:力扣454.四数相加
根据题意就是需要计算
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
暴力解法就直接4个for循环逐个遍历相加看看是否等于0即可 时间复杂度为O(n^4)
因为怎么样都要进行容器遍历,所以4个数组优解应为O(n^2)
为什么是O(n^2)?
map 存一个数组,如 A。然后计算三个数组之和,如 BCD。时间复杂度为:O(n)+O(n^3),得到 O(n^3).
map 存三个数组之和,如 ABC。然后计算一个数组,如 D。时间复杂度为:O(n^3)+O(n),得到 O(n^3).
map 存两个数组之和,如AB。然后计算两个数组之和,如 CD。时间复杂度为:O(n2)+O(n2),得到 O(n^2).
所以只需要遍历前俩个数组每个元素的和存储到map1中,将出现的次数存储到map1的值中。
再遍历后两个数组每个元素的和,用0 减去这个和 的值(反推得到map1某个元素的值 + 这个和等于0),到map1里面去找,如果有,则代表map1的值 + 这个和等于0.
那么count就等于count +上这个map1target
全代码:
class Solution {
public:int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {unordered_map<int,int> map1;int count = 0;for(int i : nums1){for(int j : nums2){map1[i+j]++;}}for(int i : nums3){for(int j : nums4){int target = 0 - (i+j);if(map1.find(target) != map1.end() ){count += map1[target];}}}return count;}
};