力扣454.四数相加

文章讲述了如何优化计算四个整数数组中满足条件nums1[i]+nums2[j]+nums3[k]+nums4[l]=0的组合数量,通过使用unordered_map减少遍历次数,从O(n^4)提升至O(n^2)的高效解决方案。
摘要由CSDN通过智能技术生成

原题链接:力扣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;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值