Leetcode18-算术三元组的数目(2367)

1、题目

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

i < j < k ,

nums[j] - nums[i] == diff 且

nums[k] - nums[j] == diff

返回不同 算术三元组 的数目。

示例 1:

输入:nums = [0,1,4,6,7,10], diff = 3

输出:2

解释:

(1, 2, 4) 是算术三元组:7 - 4 == 3 且 4 - 1 == 3 。

(2, 4, 5) 是算术三元组:10 - 7 == 3 且 7 - 4 == 3 。

示例 2:

输入:nums = [4,5,6,7,8,9], diff = 2

输出:2

解释:

(0, 2, 4) 是算术三元组:8 - 6 == 2 且 6 - 4 == 2 。

(1, 3, 5) 是算术三元组:9 - 7 == 2 且 7 - 5 == 2 。

提示:

3 <= nums.length <= 200

0 <= nums[i] <= 200

1 <= diff <= 50

nums 严格 递增。

2、解

对nums进行遍历,每个三元组即为{num[i]. nums[i] + diff, nums[i] + 2*diff}

寻找其是否存在每个元素对应的三元组中的后两个元素, 如果存在结果+1。

cpp 复制代码
    int arithmeticTriplets(vector<int> &nums, int diff)
    {
        int result = 0;
        for(int i = 0; i < nums.size(); i++){
            vector<int>::iterator mid = find((nums.begin() + i), nums.end(), nums[i] + diff);
            vector<int>::iterator last = find((nums.begin() + i + 1), nums.end(), nums[i] + 2*diff);
            if(mid != nums.end() && last != nums.end()){
                result++;
            }
        }
        return result;
    }

也可以先将nums通过哈希集合存储后进行查找

cpp 复制代码
    int arithmeticTriplets(vector<int>& nums, int diff) {
        unordered_set<int> hashSet;
        for (int x : nums) {
            hashSet.emplace(x);
        }
        int ans = 0;
        for (int x : nums) {
            if (hashSet.count(x + diff) && hashSet.count(x + 2 * diff)) {
                ans++;
            }
        }
        return ans;
    }
相关推荐
CoovallyAIHub11 分钟前
一夜之间,大模型处理长文本的难题被DeepSeek新模型彻底颠覆!
深度学习·算法·计算机视觉
再睡一夏就好16 分钟前
【C++闯关笔记】STL:deque与priority_queue的学习和使用
java·数据结构·c++·笔记·学习·
天选之女wow17 分钟前
【代码随想录算法训练营——Day43(Day42周日休息)】动态规划——300.最长递增子序列、674.最长连续递增序列、718.最长重复子数组
算法·leetcode·动态规划
我是华为OD~HR~栗栗呀37 分钟前
华为OD-23届考研-测试面经
java·c++·python·华为od·华为·面试·单元测试
敲代码的嘎仔1 小时前
JavaWeb零基础学习Day4——Maven
java·开发语言·学习·算法·maven·javaweb·学习方法
Qt程序员1 小时前
基于原子操作的 C++ 高并发跳表实现
c++·线程·c/c++·原子操作·无锁编程
聪明的笨猪猪1 小时前
hot 100 (1)—— 两数之和(哈希)
java·经验分享·算法·哈希算法
_dindong2 小时前
牛客101:链表
数据结构·c++·笔记·学习·算法·链表
wuk9982 小时前
基于位置式PID算法调节PWM占空比实现电机转速控制
单片机·嵌入式硬件·算法
不到满级不改名2 小时前
EM算法 & 隐马尔可夫模型
算法