532. 数组中的K-diff数对

 

 

 

思路：

想到：绝对值应是非负数，所以k<0的情况下要排除（测试用例有k<0的）；
设置计数器ans=0；
1、转set去重，再转回list，将newnums[]升序排列；
2、若k==0，则 2-diff 数对是相同元素对，所以遍历newnums,其中元素nums.count(newnums[i]) >= 2，
　　ans加1；
3、若k>0，则 2-diff 数对是不同元素对，遍历newnums,若(newnums[i] + k)也在newnums中，
　　则ans加1；
4、返回ans。

class Solution(object):
    def findPairs(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        # 绝对值应是非负数
        if k < 0:
            return 0
        # 转set去重，再升序排
        newnums = sorted(set(nums))
        ans = 0
        # 差为0，则是相同元素对
        if k == 0:
            for i in range(len(newnums)):
                if nums.count(newnums[i]) >= 2:
                    ans += 1
        # 差的绝对值大于0，则是不同元素对
        else:
            for i in range(len(newnums)):
                if newnums[i] + k in newnums:
                    ans += 1
        return ans


if __name__ == '__main__':
    solution = Solution()
    print(solution.findPairs([1, 3, 1, 4, 5], 0))