Leetcode 454. 四数相加 II （HashMap优化，分两组）

最新推荐文章于 2024-08-23 20:55:30 发布

wwxy261

最新推荐文章于 2024-08-23 20:55:30 发布

阅读量105

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/116718514

版权

算法专栏收录该内容

3633 篇文章 122 订阅

订阅专栏

直接做O(N^4)

双指针比较麻烦，因为有四个数组

把一个数组预处理到HashMap里，O(n^3)

把两个数组之和预处理到HashMap里，时间复杂度O(n^2)

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> cnt;
        for(auto a : A){
            for(auto b : B){
                cnt[a + b]++;
            }
        }
        int res = 0;
        for(auto c : C){
            for(auto d : D){
                res += cnt[-c - d];
            }
        }
        return res;
    }
};