[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;
}