Leetcode刷题笔记——1995. 统计特殊四元组【哈希表的应用】

最新推荐文章于 2025-05-22 19:41:20 发布

高木同学丶

最新推荐文章于 2025-05-22 19:41:20 发布

阅读量349

点赞数

分类专栏： hashmap LeetCode 文章标签： leetcode 散列表哈希算法

本文链接：https://blog.youkuaiyun.com/weixin_43904469/article/details/122223746

版权

LeetCode 同时被 2 个专栏收录

20 篇文章

订阅专栏

hashmap

3 篇文章

订阅专栏

题目

在这里插入图片描述

思路

在这里插入图片描述
a+b+c=d可以转化为a+b=d-c，且a<b<c<d，那么可以将数组分为两部分，右边部分为cd，左边部分为ab。
取一条分界线，该分界线位于右侧倒数第二个元素前，即意味着此时cd都只有一个选择，固定c为分界线的后一个元素，则d在c+1~n之间遍历，b在c-1~1之间遍历，a在1~b之间遍历。
用哈希表存下nums[d]-nums[c]出现的次数，随后在左侧遍历时统计nums[a]+nums[b]出现的次数，将此次数累加即可得到结果

代码

class Solution {
public:
    int countQuadruplets(vector<int>& nums) {
        int n = nums.size();
        unordered_map<int,int> hash;
        int res=0;

        for(int b=n-3;b>0;b--){ // 留出两个空位给cd
            int c = b+1;
            for(int d = c+1;d<n;d++){
                hash[nums[d]-nums[c]]++;
            }
            for(int a = 0;a<b;a++){
                res += hash[nums[a]+nums[b]];
            }
        }
        return res;
    }
};