力扣-每日一题20201127-454. 四数相加 II

最新推荐文章于 2024-03-29 18:08:56 发布

lc520xyp

最新推荐文章于 2024-03-29 18:08:56 发布

阅读量170

点赞数 1

本文链接：https://blog.csdn.net/qq_40967961/article/details/110246685

版权

力扣刷题专栏收录该内容

36 篇文章 0 订阅

订阅专栏

454. 四数相加 II

思路

刚开始想的是固定3个数，然后遍历一个数，如果符合条件的话，就计一次数，后来发现这样不行，那不成n4的时间复杂度了吗

于是想到二分的方法，降低一半的复杂度，变成n2，具体的做法是将AB可能组成的数存在一起，将CD可能组成的数存在一起，然后寻找互为相反数的个数

代码

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int ,int> ab;
        int res = 0;
        for(int a:A) //遍历A、B，将可以组合的数放到map中，没有将创造一个对应关系，有就将数量+1
            for(int b:B){
                ab.count(a+b)==0?ab[a+b]=1:ab[a+b]++;
            }
        for(int c:C) // 遍历C、D，看组合的相反数是否存在，存在就将res加上map的值
           for(int d:D){
               if(ab.find((c+d)*-1)!=ab.end()){
                   res += ab[(c+d)*-1];
               }
           } 
        return res;     
    }
};