C ++中数组中的K-diff对
因此,如果输入像[3,1,4,1,5],k=2,则输出将为2,因为在类似数组(1,3)和(3,5)。
为了解决这个问题,我们将遵循以下步骤-
定义称为见过的和完成的映射
定义一组
如果k<0,则-
返回0
对于初始化i:=0,当i<nums大小时,更新(将i增加1),执行-
(将看到的[nums[i]]增加1)
在s中插入nums[i]
回答:=0
对于s中的每个元素,执行-
完成[it]增加1
如果(it+k)可见但未完成,则-
如果(it-k)可见但未完成,则-
(将ans增加1)
(将ans增加1)
如果看到[它]>1,则-
(将ans增加1)
如果k等于0,则-
除此以外
返回ans
例
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h&g;
using namespace std;
class Solution {
public:
int findPairs(vector<int>& nums, int k) {
map<int, int> seen, done;
set<int> s;
if (k < 0)
return 0;
for (int i = 0; i < nums.size(); i++) {
seen[nums[i]]++;
s.insert(nums[i]);
}
int ans = 0;
for (auto it = s.begin(); it != s.end(); it++) {
if (k == 0) {
if (seen[*it] > 1)
ans++;
}
else {
done[*it]++;
if (seen.find(*it + k) != seen.end() && done.find(*it + k) == done.end())
ans++;
if (seen.find(*it - k) != seen.end() && done.find(*it - k) == done.end())
ans++;
}
}
return ans;
}
};
main(){
Solution ob;
vector<int> v = {3,1,4,1,5};
cout << (ob.findPairs(v, 2));
}输入值
{3,1,4,1,5}, 2输出结果
2