LeetCode 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) 。

• 难度：Medium

这里做题

思路

1. 显然可以用一个哈希表记录已经遍历过的元素，后面查找就可以$O(1)$时间；哈希表还可以同时完成去重的功能
2. 从前往后遍历的时候，当前为j下标，在哈希表中找i下标。注意去绝对值，所以有两种情况
3. 对于当前元素，如果可以在哈希表visited中找到组成k-diff数对的元素，那么就把较小值加入哈希表res；判断完当前元素后，当前元素加入visited。最后res的大小即为k-diff数对的数量

代码

class Solution {
    public int findPairs(int[] nums, int k) {
    	// HashSet实现Set接口
        Set<Integer> visited = new HashSet<Integer>();  // visited记录已经遍历过的元素，达到O(n)时间
        Set<Integer> res = new HashSet<Integer>();	// res为拥有k-diff数对的元素集合
        for (int num : nums) {
            if (visited.contains(num - k)) {    // contains方法，是否包含key
                res.add(num - k);   // add方法，向Set添加元素
            }
            if (visited.contains(num + k)) {
                res.add(num);
            }
            visited.add(num);
        }
        return res.size();
    }
}