2367. 算术三元组的数目

目录

一.题目描述

二.问题难点

三.解决方案

四.解决代码

五.算法的复杂度分析

---

一.题目描述

给你一个下标从 0 开始、严格递增的整数数组 nums 和一个正整数 diff 。

如果满足下述全部条件，则三元组 (i, j, k) 就是一个 算术三元组 ：

(1) i < j < k ；
(2) 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 。

二.问题难点

(1) 如何保证所取到的i,j,k满足条件i<j<k;

(2) 对满足情况的结果如何进行计数；

三.解决方案

整体思想:暴力破解法

(1) 通过三个for循环进行遍历nums取值,并通过各个控制变量的初始值来进行顺序的控制例如:

         若要保证i<j则j从i+1开始取值；

         若要保证j<k则k从j+1开始取值;

注意:在保证i<j<k条件的同时也要控制i,j,k的边界值,例如:

         i的下界min为0,   上界max为length-3;  (因为有j,k在后面不能使数组溢出)

         j的下界min为i+1,上界max为length-2 ; (因为有k在后面不能使数组溢出)

         k的下界min为j+1,上界max为length-1; (数组的长度为length不能使数组溢出)

(2)设置一个计数器变量count令其初值等于0,一旦有满足结果的情况则计数器count++；

四.解决代码

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

五.算法的复杂度分析

(1)时间复杂度：

    O(n^3).其中 n是数组nums的长度,使用三重循环暴力枚举需要O(n^3)的时间。

(2)空间复杂度：

    O(1).只需要常数的额外空间。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/number-of-arithmetic-triplets