LeetCode532-数组中的K-diff数对

今晚美团笔试

希望不会再被挂掉

王兴大哥放我一马

可还好！

---

题目描述：

给定一个整数数组和一个整数 k, 你需要在数组里找到不同的 k-diff 数对。这里将 k-diff 数对定义为一个整数对 (i, j), 其中 i 和 j 都是数组中的数字，且两数之差的绝对值是 k.

示例 1:

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

示例 2:

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

示例 3:

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

思路解析：

这题我会提供两种解法，第一种是双指针法，第二种算是双set集合法吧。

方法一：双指针法

很简单，首先对给定数组nums排序，然后就是定义首尾指针start和end，while循环一次遍历数组。此题的另一关键点就是k=0的情况了，这涉及到指针start和end指向的数字相同。可能我这么说你觉得不麻烦，等会儿你看代码就知道了情况有些多吧，因为这个是一次遍历而不是两次暴力法遍历，所以考虑的情况要多些。言归正传，碰到指针start和end指向的数字相同的情况，那就是把start和end整体右移一位。另外，我们的数组没有去重，start或end指针右移时可能会碰到相同的数字，此时也需判断一下，如果相同那就把相应指针右移一位。最后就是start指针的下标值肯定不能超过end指针对应的下标值啦。

执行过程如下所示：

代码如下：

class Solution(object):
    def findPairs(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        if len(nums) == 0:
            return 0
        nums.sort()
        start = 0
        end = 1
        pairs_num = 0
        flag = nums[0] - 1
        while end < len(nums):
            if start == end:
                end += 1
                continue
            if nums[start] == flag:
                start += 1
                if nums[end] == flag:
                    end += 1
                continue
            if nums[end] - nums[start] == k:
                pairs_num += 1
                flag = nums[start]
                start += 1
                end += 1
            elif nums[end] - nums[start] > k:
                start += 1
            elif nums[end] - nums[start] < k:
                end += 1
        return pairs_num

if __name__ == "__main__":
    nums = [1]
    k = 2
    pairs_num = Solution().findPairs(nums, k)
    print(pairs_num)

执行效率在50%左右吧，中等偏下了。

方法二：双set集合法

这个方式实属巧妙了，不用sort排序数组，考虑的情况也不多。真的6啊，就是看你的脑子够不够用了，我先说我不够用哈哈哈哈。回到正题上来，首先是定义两个num_set, abs_set去重set集合，其中，num_set用来保存每次遍历时nums数组中的去重值，可以理解为给nums数组去重。abs_set用来保存当前遍历的元素与k的差值，因为是只要差值的绝对值在给定nums数组中即可。所以得分两种情况，即i+k和i-k。最后我们只需要看abs_set集合中有多少元素，就说明我们有多少对符合要求的数字了。是不是很巧妙，真的是太妙了。

代码如下：

class Solution(object):
    def findPairs(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        if k < 0:
            return 0
        num_set, abs_set = set(), set()
        for num in nums:
            if num-k in num_set:
                abs_set.add(num-k)
            if num+k in num_set:
                abs_set.add(num)
            num_set.add(num)
        return len(abs_set)


if __name__ == "__main__":
    nums = [1]
    k = 2
    pairs_num = Solution().findPairs(nums, k)
    print(pairs_num)

但是执行效率好像并没有较方法一好太多，害，不懂这测分系统！