力扣454.四数相加

原题链接:力扣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;
    }
};

你可能感兴趣的:(leetcode,算法)