从排序数组中计算三元组，相邻元素之间的差异等于 D

简介

这是一个用于计算排序数组中的三元组的算法。该算法的输入是一个已排序的数组，要求计算出数组中所有差异等于给定差值 D 的三元组。三元组是由相邻的三个元素组成，且它们之间的差异等于给定的 D。

算法思路

1. 遍历数组，从第二个元素开始。
2. 对于每个元素 nums[i]，查找 nums[i]-D 和 nums[i]+D 是否在数组中。可以使用二分搜索来提高查找的效率。
3. 如果找到了符合条件的三个数，将其存储在结果数组中。
4. 遍历完整个数组后，返回结果数组。

伪代码

def findTriplets(nums, D):
    result = []
    
    for i in range(1, len(nums)):
        target1 = nums[i] - D
        target2 = nums[i] + D
        
        if binarySearch(nums, target1) and binarySearch(nums, target2):
            result.append((nums[i] - D, nums[i], nums[i] + D))
    
    return result

def binarySearch(nums, target):
    left = 0
    right = len(nums) - 1
    
    while left <= right:
        mid = (left + right) // 2
        
        if nums[mid] == target:
            return True
        
        if nums[mid] < target:
            left = mid + 1
        else:
            right = mid - 1
    
    return False

复杂度分析

• 时间复杂度：主要消耗时间的操作是二分搜索，在最坏情况下，需要对每个元素进行一次二分搜索，因此时间复杂度为 O(n log n)，其中 n 是数组的长度。
• 空间复杂度：除了存储结果的数组外，算法的空间复杂度为 O(1)。