LeetCode 532. 数组中的 k-diff 数对

532. 数组中的 k-diff 数对

给你一个整数数组 nums 和一个整数 k，请你在数组中找出 不同的 k-diff 数对，并返回不同的 k-diff 数对 的数目。

k-diff 数对定义为一个整数对 (nums[i], nums[j]) ，并满足下述全部条件：

• 0 <= i, j < nums.length
• i != j
• |nums[i] - nums[j]| == k

注意，|val| 表示 val 的绝对值。

示例 1：

输入：nums = [3, 1, 4, 1, 5], k = 2
输出：2
解释：数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
尽管数组中有两个 1 ，但我们只应返回不同的数对的数量。

示例 2：

输入：nums = [1, 2, 3, 4, 5], k = 1
输出：4
解释：数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5) 。

示例 3：

输入：nums = [1, 3, 1, 5, 4], k = 0
输出：1
解释：数组中只有一个 0-diff 数对，(1, 1) 。

提示：

• 1 <= nums.length <= 10^4
• -10^7 <= nums[i] <= 10^7
• 0 <= k <= 10^7

解法： 两数之和思想
 

两数之和 的 同类题型：

LeetCode 1. 两数之和-优快云博客

LeetCode 2563. 统计公平数对的数目-优快云博客

LeetCode 2824. 统计和小于目标的下标对数目-优快云博客

 LeetCode 1248. 统计「优美子数组」-优快云博客

LeetCode 930. 和相同的二元子数组-优快云博客

LeetCode 2006. 差的绝对值为 K 的数对数目-优快云博客

class Solution {
    public int findPairs(int[] nums, int k) {
        int n = nums.length;
        // ans 存储min(nums[i],nums[j]), 以避免答案重复
        Set<Integer> ans = new HashSet<>();
        // set 存储nums[j]
        Set<Integer> set = new HashSet<>();
        set.add(nums[0]);
        for (int j = 1; j < n; j++) {
            // 0 <= i < j < n, |nums[i] - nums[j] | == k
            // 枚举j，找到满足 nums[i] = nums[j] + k 或 nums[i] = nums[j] - k 的nums[i]的数目
            // 计算不同的 k-diff 数对 的数目,指 nums[i] 是唯一的,不能重复
            if (set.contains(nums[j] + k)) {
                // nums[i] == nums[j] + k, nums[i] > nums[j]
                ans.add(nums[j]);
            }
            if (set.contains(nums[j] - k)) {
                // nums[i] == nums[j] - k, nums[i] < nums[j]
                ans.add(nums[j] - k);
            }
            set.add(nums[j]);
        }
        return ans.size();
    }
}

复杂度分析 

• 时间复杂度：O(n)，n 是 数组nums 的长度。
• 空间复杂度：O(n), n 是 数组nums 的长度。最坏情况下，set 中元素的数目和 nums 中元素数目相同。