力扣 454. 四数相加 II

江湖人称贺行风

已于 2023-04-13 11:33:49 修改

阅读量63

点赞数

分类专栏：开始C++吧文章标签： leetcode 算法 c++ 哈希表数据结构

于 2023-04-13 11:14:58 首次发布

本文链接：https://blog.csdn.net/u010950122/article/details/130125656

版权

开始C++吧专栏收录该内容

176 篇文章 1 订阅

订阅专栏

题目来源：https://leetcode.cn/problems/4sum-ii/description/

C++题解：

1）暴力解法：用4个for循环，暴力求解能得到正确答案，但是时间复杂度为 $O(n^4)$ ，力扣过不了。

2）哈希表：将4个数组两两看待，解题思路与力扣 1. 两数之和相似。

使用unordered_map存放，key值为nums1和nums2元素和的值，value为nums1和nums2元素和的出现次数，这样使用2个for循环就能完成unordered_map的存放；

为寻求4个数组和为0，在unordered_map中寻找（0 - nums3[k] - nums[4]）是否出现及出现次数，将出现次数加到结果中。这一步也用到2个for循环。

整体时间复杂度为 $O(2*n^2) = O(n^2)$ 。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        int len = nums1.size();
        int res = 0;
        unordered_map<int, int> umap;
        for(int ii = 0; ii < len; ii++) {
            for(int jj = 0; jj < len; jj++) {
                //可直接写umap[nums1[ii] + nums2[jj]]++;
                auto iter = umap.find(nums1[ii] + nums2[jj]);
                if(iter != umap.end()) iter->second++;
                else umap.insert(pair<int, int>(nums1[ii] + nums2[jj], 1));
            }
        }
        for(int kk = 0; kk < len; kk++) {
            for(int ll = 0; ll < len; ll++) {
                auto iter = umap.find(0 - (nums3[kk] + nums4[ll]));
                if(iter != umap.end()) {
                    res += iter->second;
                }
            }
        }
        return res;
    }
};