LeetCode——2367. 算是三元组的数目

通过万岁！！！

• 题目：给你一个严格递增的数组nums，然后让你找三个数，下标分别为i、j、k，并且i<j<k。前一个减去后一个刚好等于diff，则表示这一组i、j、k满足。让你找这个递增数组中有多少个。
• 思路1：首先就是把数组转成set，然后判断nums[i]+diff是不是在set中并且nums[i]+diff+diff是不是在。如果都在就满足。这样的时间复杂度是2*n，空间复杂度是n。
• 思路2：上面的思路，我们其实没有用到严格有序这一点。思路2就是以i为标杆，然后让j快速跳过nums[j]<nums[i]+diff的这一部分，然后j跳完之后，然后跳k，其条件也是跳过nums[k]<nums[j]+diff的一部分。最后判断一下，现在i、j、k是不是满足，如果满足，则个数加1，然后j、k都加1。但是不管满不满足i都需要+1。并且如果j和k在进行跳跃的时候，如果一个到达了尾部，那么就没必要进行了。这样的空间复杂度是0，然后分析一下时间复杂度，我们这里用k进行分析，其实k只是针对数组遍历了一遍，因为k是最右侧的指针，二j和i指针也不会回头，所以时间复杂度是n。
• 技巧：哈希、双指针

java代码——思路1

class Solution {
    public int arithmeticTriplets(int[] nums, int diff) {
        Set<Integer> set = Arrays.stream(nums).boxed().collect(Collectors.toSet());
        int res = 0;
        for (int i : nums) {
            if (set.contains(i + diff) && set.contains(i + diff + diff)) {
                res++;
            }
        }
        return res;
    }
}

java代码——思路2

class Solution {
    public int arithmeticTriplets(int[] nums, int diff) {
        int res = 0, i = 0, j = 1, k = 2, n = nums.length;
        while (k < n) {
            // 先找j
            while (nums[j] - nums[i] < diff) {
                if (j == n - 1)// 最后一个了
                    return res;
                j++;
            }
            // 再找k
            while (nums[k] - nums[j] < diff) {
                if (k == n - 1)// 最后一个了
                    return res;
                k++;
            }
            if (nums[j] - nums[i] == diff && nums[k] - nums[j] == diff) {
                res++;
                j++;
                k++;
            }
            i++;
        }
        return res;
    }
}

• 总结：题目还是比较有意思的，特别是思路2，虽然有两层循环，但是时间复杂度却是n。