【easy】532. K-diff Pairs in an Array
这道题给了我们一个含有重复数字的无序数组,还有一个整数k,让我们找出有多少对不重复的数对(i, j)使得i和j的差刚好为k。由于k有可能为0,而只有含有至少两个相同的数字才能形成数对,那么就是说我们需要统计数组中每个数字的个数。我们可以建立每个数字和其出现次数之间的映射,然后遍历哈希表中的数字,如果k为0且该数字出现的次数大于1,则结果res自增1;如果k不为0,且用当前数字加上k后得到的新数字也在数组中存在,则结果res自增1,参见代码如下:
/* 思路:因为存在k=0的情况,并且答案需要是完全不重复的数对,所以 可以先存下每个数字出现的次数,如果k=0且每个数字出现次数大于1,则count++ 如果k>0,且当前数字+k之后的数字也出现过,那么count++ */ class Solution { public: int findPairs(vector<int>& nums, int k) { int count = 0; int len = nums.size(); unordered_map<int,int> map; for (auto num:nums) map[num]++; for (auto a:map){ if (k==0 && a.second>1) count++; if (k>0 && map.count(a.first + k)) count++; } return count; } };
***一定要学会unordered_map,auto等,这个很好写哦