LeetCode 每日一题 2020-11-27(四数相加 II)

最新推荐文章于 2022-03-09 11:08:18 发布

星河有鱼

最新推荐文章于 2022-03-09 11:08:18 发布

阅读量203

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/Baimax1/article/details/110236460

版权

LeetCode 专栏收录该内容

94 篇文章 0 订阅

订阅专栏

454. 四数相加 II

难度中等

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。

为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -228 到 228 - 1 之间，最终结果不会超过 231 - 1 。

思路：

相比于15. 三数之和、 18. 四数之和，本题不需要考虑是否重复的问题，因此只需要判断最后四数之和为零即可。

考虑每个数组的不同字符组合，若是简单的直接多重循环，时间复杂度为 O(n^4)，且组合数需要的空间可能为 O(n^3)。

选择使用两个数组相组合实现，后两个数组的某一组合与前一个数组的组合数和为零，次数加1，空间复杂度均为 O(n^2)。

int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
    unordered_map<int,int> data;
    int len=A.size();
    for(auto& a:A){
        for(auto& b:B){
            data[a+b]++;
        }
    }
    int ans=0;
    for(auto& c:C){
        for(auto& d:D){
            if(data.count(-c-d))
                ans+=data[-c-d];
        }
    }
    return ans;
}

进阶思路：

组合数可以借助乘法运算（ a中有 5个2，b中有6个1，不考虑其他情况下，a、b的和至少有 30种情况为 3 ）；

这种思路针对数组中内容完全一致的情况可以，节约剪除额外组合次数。

int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
    unordered_map<int,int> s1,s2,s3,s4,data;
    for(auto& a:A) s1[a]++;
    for(auto& b:B) s2[b]++;
    for(auto& c:C) s3[c]++;
    for(auto& d:D) s4[d]++;
    for(auto& a:s1){
        for(auto& b:s2){
            data[a.first + b.first] += a.second * b.second;
        }
    }
    int ans=0;
    for(auto& c:s3){
        for(auto& d:s4){
            if(data.count(-(c.first + d.first) ))
                ans += c.second * d.second * data[-(c.first + d.first)];
        }
    }
    return ans;
}