LeetCode 2367.算术三元组的数目（枚举 || 哈希表）

最新推荐文章于 2024-09-16 06:54:08 发布

岁忧

最新推荐文章于 2024-09-16 06:54:08 发布

阅读量176

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展 c++ 枚举哈希表

本文链接：https://blog.csdn.net/weixin_46028214/article/details/139022738

版权

LeetCode 专栏收录该内容

458 篇文章 0 订阅

订阅专栏

2367.算术三元组的数目

在这里插入图片描述
思路：方法一，三层for循环来枚举。

class Solution {
public:
    int arithmeticTriplets(vector<int>& nums, int diff) {
        int n=nums.size();
        if(n<3) return 0;
        int ct=0;
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                for(int k=j+1;k<n;k++){
                    if(nums[j]-nums[i]==diff &&nums[k]-nums[j]==diff) ct++;
                }
            }
        }
        return ct;
    }
};

思路：方法二，用一层for循环来预处理出当前位置i，前面有多少值为nums[i]-diff的元素。同理，找到后面有多少值为nums[i]+diff的元素。用哈希表来记录，然后两个值相乘，答案就出来了

class Solution {
public:
    int arithmeticTriplets(vector<int>& nums, int diff) {
        int n=nums.size();
        if(n<3) return 0;
        unordered_map<int ,int> mp1,mp2;
        vector<int> v1(n,0),v2(n,0);
        for(int i=0;i<n;i++){
            if(mp1.find(nums[i]-diff)!=mp1.end()){
                v1[i]=mp1[nums[i]-diff];
            }
            mp1[nums[i]]++;
        }
        for(int i=n-1;i>=0;i--){
            if(mp2.find(nums[i]+diff)!=mp2.end()){
                v2[i]=mp2[nums[i]+diff];
            }
            mp2[nums[i]]++;
            //cout<<v1[i]<<" "<<v2[i]<<endl;
        }
        int ct=0;
        for(int i=1;i<n-1;i++){
            ct+=v1[i]*v2[i];
        }
        return ct;
    }
};