找到元素X的数量，使得X + K也存在于数组中

简介

在一个给定的数组中，需要找到元素X的数量，使得X + K也存在于数组中。K是一个常数，X和X + K必须在数组中不同。

示例

输入: arr = [1,2,3,4,5], k = 1
输出: 4
解释: 在数组中存在4对(X, X+K)：(1,2),(2,3),(3,4),(4,5)

实现思路

• 将数组中的所有元素存储到一个set集合中，便于后续的查找和去重操作
• 遍历数组中的每个元素X，判断X+K是否存在于数组中，如果存在则计数器+1
• 遍历完成后返回计数器的值即为答案

代码实现

以下是Python语言的示例代码：

def count_pairs(arr, k):
    # 将所有元素存储到集合中，去重并快速查找元素
    nums = set(arr)
    count = 0
    for num in arr:
        # 如果X+K存在于集合中则计数器+1
        if (num + k) in nums:
            count += 1
    return count

复杂度分析

• 时间复杂度：O(n), 遍历了一遍数组
• 空间复杂度：O(n), 存储了一个set集合