LeetCode454 四数相加 II

红毛乌龟

已于 2024-07-27 15:37:07 修改

阅读量282

点赞数 4

分类专栏：算法刷题文章标签：算法数据结构 c++ leetcode 哈希表

于 2024-07-27 15:34:52 首次发布

本文链接：https://blog.csdn.net/daishabby2486/article/details/140736163

版权

算法刷题专栏收录该内容

90 篇文章 0 订阅

订阅专栏

前言

题目： 454. 四数相加 II
文档： 代码随想录——四数相加 II
编程语言： C++
解题状态： 思路错误

思路

四个数组简单地使用 $f or$ 循环的话，时间复杂度就太高了，一种有效的方法就是化四为二，两两形成一个整体，再进行简单的处理。

代码

哈希表

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        std::unordered_map<int, int> umap;

        int res = 0;

        for (int a : nums1) {
            for (int b : nums2) {
                umap[a + b]++;
            }
        }

        for (int c : nums3) {
            for (int d : nums4) {
                if (umap.find(0 - c - d) != umap.end()) {
                    res += umap[0 - c - d];
                }
            }
        }

        return res;
    }
};