LeetCode #532 K-diff Pairs in an Array

最新推荐文章于 2018-10-27 23:05:10 发布

Jack_CJZ

最新推荐文章于 2018-10-27 23:05:10 发布

阅读量199

点赞数

分类专栏： LeetCode 文章标签： leetcode 哈希表

本文链接：https://blog.csdn.net/Jack_CJZ/article/details/77985849

版权

LeetCode 专栏收录该内容

20 篇文章 0 订阅

订阅专栏

题目

Given an array of integers and an integer k, you need to find the number of unique k-diff pairs in the array. Here a k-diff pair is defined as an integer pair (i, j), where i and j are both numbers in the array and their absolute difference is k.

Example 1:

Input: [3, 1, 4, 1, 5], k = 2
Output: 2
Explanation: There are two 2-diff pairs in the array, (1, 3) and (3, 5).
Although we have two 1s in the input, we should only return the number of unique pairs.

Example 2:

Input:[1, 2, 3, 4, 5], k = 1
Output: 4
Explanation: There are four 1-diff pairs in the array, (1, 2), (2, 3), (3, 4) and (4, 5).

Example 3:

Input: [1, 3, 1, 5, 4], k = 0
Output: 1
Explanation: There is one 0-diff pair in the array, (1, 1).

Note:

The pairs (i, j) and (j, i) count as the same pair.
The length of the array won’t exceed 10,000.
All the integers in the given input belong to the range: [-1e7, 1e7].

解题思路

我在解这道题时，主要踩进了三个坑：

结果中的 k-diff pairs 不允许重复。比如 example1 的数组中，含有两个 1 ，这样，在遍历所有的结果后会得到 (1, 3), (1, 3) and (3, 5) ，但由于不允许重复，所以只能有一个 (1, 3) 。
当 k == 0 时又必须要考虑重复，因为只有重复的数字才能组成满足条件的 k-diff pair 。
当 k < 0 时，无解，返回 0 。

针对以上三个坑，最后得出的解题思路如下：因为 (i, j) 和 (j, i) 视为相同，因此可以创建一个集合，当扫描得到 (i, j) 时，把 i 和 j 中较大的那一个数存入集合中，代表 (i, j) 和 (j, i) 两个 k-diff pairs 都已经存进集合中了，这样，当扫描得到 (j, i) 时，由于i 和 j 中较大的那一个数已经在集合中，则不再需要添加了，这样就可以解决上面的坑了，最后返回集合的元素个数即可。需要创建两个集合，其中一个用于保存结果，另一个用于实现哈希查找。

Java代码实现

class Solution {
    public int findPairs(int[] nums, int k) {
        Set<Integer> rstPairs = new HashSet<>();
        Set<Integer> hash = new HashSet<>();

        if (k < 0) { return 0; }

        for (int num : nums) {
            if (hash.contains(num - k)) {
                if (!rstPairs.contains(num))
                    rstPairs.add(num);
            }
            if (hash.contains(num + k)) {
                if (!rstPairs.contains(num + k))
                    rstPairs.add(num + k);
            }

            if (!hash.contains(num))
                hash.add(num);
        }

        return rstPairs.size();
    }
}