题目要求找出数组中差值为k的对数,要求有重复元素出现时不可重复计数。
两种思路求解:
1. 先将数组排序,再设定指针定位元素位置。NOTE:使用while循环排除掉重复元素。
class Solution {
public:
int findPairs(vector<int>& nums, int k) {
if(k<0) return 0;
int res = 0;
sort(nums.begin(), nums.end());
for(int i=0, j=1, n=nums.size(); i<n && j<n;)
{
if(i==j || nums[j]-nums[i]<k) j++;
else if(nums[j]-nums[i]>k) i++;
else{
res++;
while(j<n && nums[j]-nums[i]==k) j++;
while(j<n-1 && nums[j]==nums[j+1]) j++;
int ii = i;
if(i<j && nums[ii]==nums[i]) i++;
}
}
return res;
}
};
2. 使用map记录元素键值,通过判断n+k、n-k是否在map中出现过改变返回元素res的值(即差为k的数目)。
class Solution {
public:
int findPairs(vector<int>& nums, int k) {
if(k<0) return 0;
int res = 0;
unordered_map<int, int> my;
for(int &n:nums)
{
if(my.count(n-k) && !my[n-k]) ++res, ++my[n-k];
if(my.count(n+k) && !my[n]) ++res, ++my[n];
if(!my.count(n)) my[n]=0;
}
return res;
}
};
最近做题有点拖延,遇到不会的点立马去找不要等。方法一的指针定位元素位置的题做过好几次了,仍然不能熟练运用。中间停顿的时间太长,题感会变差。每天做几道,逐渐找回状态。