[LeetCode]532. K-diff Pairs in an Array

最新推荐文章于 2024-09-29 23:23:41 发布

charon____

最新推荐文章于 2024-09-29 23:23:41 发布

阅读量424

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/Lcharon/article/details/72871726

版权

leetcode 专栏收录该内容

184 篇文章 0 订阅

订阅专栏

[LeetCode]532. K-diff Pairs in an Array

题目描述

这里写图片描述

思路

哈希，计数
如果k为负，直接返回0
如果k为0，即为求重复元素

代码

#include <iostream>
#include <unordered_map>
#include <vector>
#include <algorithm>

using namespace std;

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        int res = 0;
        if (k < 0)
            return res;
        unordered_map<int, int> um;
        for (int num : nums)
            um[num]++;
        for (auto p : um) {
            if (k == 0) {
                if (p.second > 1) res += 1;
            }
            else {
                if (um.count(p.first + k)) res += 1;
            }
        }
        return res;
    }
};

int main() {
    vector<int> nums = { 1,2,3,4,5 };
    Solution s;

    cout << s.findPairs(nums, -1) << endl;

    system("pause");
    return 0;
}