在n个元素的数组中,找到差值为k的数字对去重后的个数
1. 暴力
本题思路:
1)二重循环枚举数对
2)把数对放在set里去重
3)输出set的大小
#include<iostream>
#include<set>
using namespace std ;
int n,k,a[100000] ;
set<pair<int , int> > mypairs ;
int main()
{
cin >> n >> k ;
for(int i=0;i<n;i++)
cin >> a[i] ;
for(int i=0;i<n;i++){
for(int j=0;j<n;j++)
if(a[i]+k==a[j])
mypairs.insert(make_pair(a[i],a[j])) ;
}
cout << mypairs.size()<<endl ;
return 0;
}
缺点是复杂度为O(N^2) N=100000时估计要超时
2.优化
#include<unordered_set>
#include<iostream>
using namespace std ;
int n , k , x ,ans =0 ;
unordered_set <int > mtset ;
int main()
{
cin >> n >> k ;
for(int i=0;i<n;i++){
cin >> x ;
myset.insert(x) ;
}
for(int x:myset){
if(myset.find(x+k)!=myset.end())
ans++ ;
}
cout << ans <<endl;
return 0;
}
#include<set>
#include<iostream>
using namespace std ;
int n , k ,x ,ans = 0;
set <int> myset;
int main()
{
cin >> n >> k ;
for(int i=0;i<n;i++){
cin >> n ;
myset.insert(x) ;
}
for(int set<int>::iterator i=myset.begin();i!=myset.end();i++ )
{
if(myset.find((*i)+k)!=myset.end())
ans++;
}
cout << ans << endl ;
return 0;
}