Leetcode 454. 四数相加 II

最新推荐文章于 2024-07-04 10:15:20 发布

人不学习就是一坨屎

最新推荐文章于 2024-07-04 10:15:20 发布

阅读量280

点赞数 5

分类专栏：代码随想录文章标签： leetcode 算法

本文链接：https://blog.csdn.net/2401_82394804/article/details/138138362

版权

代码随想录专栏收录该内容

26 篇文章 0 订阅

订阅专栏

文章讲述了如何使用哈希表优化解决四数相加II问题，通过构建哈希表存储两数之和及其出现次数，然后遍历其他两个数组寻找和为0的组合，时间复杂度和空间复杂度均为O(N^2)。

摘要由CSDN通过智能技术生成

题源: 454. 四数相加 II
题目的核心是求解四个数组 nums1, nums2, nums3, 和 nums4 中所有元素组合 (a, b, c, d) 的数量，使得它们的和为 0。这道题可以通过哈希表来优化查找和计算过程，减少时间复杂度。

解题思路

构建哈希表:
首先，遍历数组 nums1 和 nums2，计算所有可能的两数之和，并将这些和存储在一个哈希表中（以和为键，出现次数为值）。这一步是为了记录每一种可能的两数之和及其出现的次数，为后面的匹配操作提供便利。
计算匹配和:
然后，遍历数组 nums3 和 nums4，对于每对 (c, d)，计算 -(c + d)，查找这个值是否已经作为键存在于哈希表中。如果存在，说明找到了对应的 (a, b) 组合，使得 a + b + c + d = 0。将哈希表中该键的值（即 (a, b) 组合出现的次数）累加到结果 count 中。
返回结果:
遍历完成后，count 中存储的就是所有满足 a + b + c + d = 0 的四元组组合的数量。

时间复杂度

构建哈希表的时间复杂度为 O(N^2)，其中 N 是单个数组的长度（假设所有数组长度相等）。这是因为需要遍历 nums1 和 nums2 的所有元素组合。
查找和匹配的时间复杂度同样为 O(N^2)，因为需要遍历 nums3 和 nums4 的所有元素组合，并进行哈希表查找操作。

因此，整体时间复杂度为 O(N^2)。

空间复杂度

哈希表的空间复杂度最坏情况下也为 O(N^2)，这取决于 nums1 和 nums2 两数之和的不同结果数量。

Code

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> map;
        for(int a : nums1)
            for(int b : nums2)
                map[a + b] ++;
        
        int count = 0;        //记录a + b + c + d = 0的次数
        for(int c : nums3)
            for(int d : nums4)
                if(map.find(0 - (c + d)) != map.end())
                    count += map[0 - (c + d)];        
                    //map[0 - (c + d)]的值代表所有使得 a + b = -(c + d) 的 (a, b) 对的数量，不能简单写为count ++
        return count;
    }
};