leetcode[2006]差的绝对值为 K 的数对数目 python3实现(Counter计数,一次遍历)

这篇博客详细介绍了如何使用Python3的Counter方法解决LeetCode第2006题,通过一次遍历找到数组中差的绝对值为K的数对的数目。内容涵盖了问题描述、解题思路、代码实现以及可能的优化策略。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

# 给你一个整数数组 nums 和一个整数 k ,请你返回数对 (i, j) 的数目,满足 i < j 且 |nums[i] - nums[j]| == k 。
#  
# 
#  |x| 的值定义为: 
# 
#  
#  如果 x >= 0 ,那么值为 x 。 
#  如果 x < 0 ,那么值为 -x 。 
#  
# 
#  
# 
#  示例 1: 
# 
#  输入:nums = [1,2,2,1], k = 1
# 输出:4
# 解释:差的绝对值为 1 的数对为:
# - [1,2,2,1]
# - [1,2,2,1]
# - [1,2,2,1]
# - [1,2,2,1]
#  
# 
#  示例 2: 
# 
#  输入:nums = [1,3], k = 3
# 输出:0
# 解释:没有任何数对差的绝对值为 3 。
#  
# 
#  示例 3: 
# 
#  输入:nums = [3,2,1,5,4], k = 2
# 输出:3
# 解释:差的绝对值为 2 的数对为:
# - [3,2,1,5,4]
# - [3,2,1,5,4]
# - [3,2,1,5,4]
#  
# 
#  
# 
#  提示: 
# 
#  
#  1 <= nums.length <= 200 
#  1 <= nums[i] <= 100 
#  1 <= k <= 99 
#  
#  Related Topics 数组 哈希表 计数 👍 44 👎 0


# leetcode submit region begin(Prohibit modification and deletion)
class Solution:
    def countKDifference(self, nums: List[int], k: int) -> int:
        ret = 0
        cnt = Counter(nums)
        for key in cnt.keys():
            if key + k in cnt.keys():
                ret += (cnt[key+k] * cnt[key])
            if key - k in cnt.keys():
                ret += (cnt[key-k] * cnt[key])
        return ret // 2
# leetcode submit region end(Prohibit modification and deletion)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值