Bootstrap

代码随想录算法【Day7】

DAY7

454.四数相加II

特点:

1.只用返回元组的个数,而不用返回具体的元组

2.可以不用去重

暴力思路:遍历,这样时间复杂度会达到O(n^4)

标准思路:用哈希法(场景:在一个集合里面判断一个元素是否出现过)

先遍历数组A、B,统计a+b的结果以及这个结果出现的次数,然后遍历数组C、D,判断0-(c+d)有没有在a+b的集合里面出现过。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> umap; //<key, value>分别代表a+b的值,和这个值出现的次数
        for(int a : nums1)
            for(int b : nums2)
                umap[a + b] ++;
        //当访问一个不存在的键时,map 会自动为这个键创建一个新条目
        //新条目的值会被默认初始化,对于整型(int),默认初始化的值就是 0
        //所以不用去考虑给umap[a + b]设置一个初始值
        int count = 0; //最后的结果
        for(int c : nums3)
            for(int d :nums4)
                if(umap.find(0 - (c + d)) != umap.end())
                    count += umap[0 - (c + d)];
        return count;
    }
};
;