LeetCode 2006 差的绝对值为K的数对数目[Map] HERODING的LeetCode之路

最新推荐文章于 2025-10-29 13:09:05 发布

原创最新推荐文章于 2025-10-29 13:09:05 发布 · 460 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #排序算法 #c++ #map

LeetCode 专栏收录该内容

733 篇文章

订阅专栏

本文介绍了如何使用Python解决含有绝对值的计数问题，通过构建unordered_map存储每个数的出现次数，遍历并计算大于当前数k的数对乘积，最终返回答案。

在这里插入图片描述
解题思路：
既然加了绝对值，那么只需要统计比每个数大k的数对个数即可，用map记录每个数的个数，然后遍历每个map对，判断比当前数大k的数是否在map中有记录，有的话直接二者个数相乘，加到ans中，代码如下：

class Solution {
public:
    int countKDifference(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        for(int& num : nums) {
            mp[num] ++;
        }
        int ans = 0;
        for(auto& [a, b] : mp) {
            if(mp.count(a + k)) {
                ans += b * mp[a + k];
            }
        }
        return ans;
    }
};