2023-03-31 LeetCode每日一题（算术三元组的数目）

2023-03-31每日一题

一、题目编号

2367. 算术三元组的数目

二、题目链接

https://leetcode.cn/problems/number-of-arithmetic-triplets/

三、题目描述

给你一个下标从 0 开始、严格递增 的整数数组 nums 和一个正整数 diff 。如果满足下述全部条件，则三元组 (i, j, k) 就是一个 算术三元组 ：

i < j < k ，
nums[j] - nums[i] == diff 且
nums[k] - nums[j] == diff
返回不同 算术三元组 的数目。

四、解题代码

class Solution {
public:
    unordered_map<int, int> hash1;
    int arithmeticTriplets(vector<int>& nums, int diff) {
        int n = nums.size();
        for(int i = 0; i < n; ++i){
            hash1[nums[i]]++;        }
        int cnt = 0;
        for(int i = 0; i < n; ++i){
            if(hash1[nums[i]-diff] && hash1[nums[i]+diff]){
                ++cnt;
            }
        }
    return cnt;
    }
};

五、解题思路

（1）因为题目已经给出了该数组是严格递增的，意味着该数组中没有相同元素。

（2）我们先一次遍历，用哈希表来存储数组中的每一个数字。

（3）接着我们再遍历一次，每遍历一个数字，我们判断该数字是否是算术三元组中的nums[j]。如果是，则哈希表中一定存在nums[j] - diff 和 nums[j]+diff。

（4）最后返回统计的结果即可。