leetcode454 四数相加

dokii12025-03-10 13:05

四数相加Ⅱ的解法可以将四数分为两组，即"分组 + 哈希"：

初始化哈希表。

分组：nums1 和 nums2 一组，nums3 和 nums4 一组。

分别对 nums1 和 nums2 进行遍历，将所有 nums1 和 nums2 的值的和作为哈希表的 key，和的次数作为哈希表的 value。

分别对 nums3 和 nums4 进行遍历，若 -(nums3[k] + nums4[l]) 在哈希表中，则四元组次数 +hash[-(nums3[k]+nums4[l])] 次。

或者说：

首先定义一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。

遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。

定义int变量count，用来统计 a+b+c+d = 0 出现的次数。

再遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。

最后返回统计值 count 就可以了

cpp 复制代码

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> hash;
        int count = 0;
        for(int i = 0; i < nums1.size(); i++){
            for(int j = 0; j < nums2.size(); j++){
                hash[nums1[i] + nums2[j]]++;
            }
        }
        for(int i = 0; i < nums3.size(); i++){
            for(int j = 0; j < nums2.size(); j++){
                if(hash.find(-(nums3[i]+nums4[j])) != hash.end()){
                    count += hash[-(nums3[i]+nums4[j])];
                }
            }
        }
        return count;
    }
};

更简洁的一种遍历方式

cpp 复制代码

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> hash;
        int count = 0;
        for(int a: nums1){
            for(int b: nums2){
                hash[a + b]++;
            }
        }
        for(int c: nums3){
            for(int d: nums4){
                if(hash.find(-(c + d)) != hash.end()){
                    count += hash[-(c + d)];
                }
            }
        }
        return count;
    }
};