leetcode——454.四数相加II

最新推荐文章于 2025-04-26 20:37:20 发布

原创最新推荐文章于 2025-04-26 20:37:20 发布 · 87 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

145 篇文章

订阅专栏

本文介绍了一种解决四数之和问题的有效算法。通过将数组分成两组，利用哈希表统计前两组数相加的结果频次，并查找后两组数相加结果的相反数是否存在哈希表中。此方法显著提高了求解效率。

思路

将四个数组分为两组，一组用于计数，一组用于查找

代码

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
    	//将前两个数组的结果进行相加 计数
        unordered_map<int, int> countAB;
        for (int u: A) 
            for (int v: B) 
                countAB[u + v]++;
        //将后两个数组的结果进行相加 查找
        int ans = 0;
        for (int u: C) 
            for (int v: D) 
                if (countAB.count(-u - v)) 
                    ans += countAB[-u - v];
                
        return ans;
    }
};