题目描述
请设计一个高效算法,找出数组中两数之和为指定值的所有整数对。
给定一个int数组A和数组大小n以及需查找的和sum,请返回和为sum的整数对的个数。保证数组大小小于等于3000。
测试样例:
[1,2,3,4,5],5,6
返回:2
方法一:双指针法:
class FindPair {
public:
int countPairs(vector<int> A, int n, int sum) {
// write code here
sort(A.begin(),A.end());
int i=0,j=n-1;
int count=0;
int s=0;
while(i<j){
s=A[i]+A[j];
if(s==sum){
if(A[i]==A[j]){//整个数组都一样
count+=(j-i)*(j-i+1)/2;
break;
}
else{//防止有重复的数
int cl=i,cr=j;
do
i++;
while(A[i]==A[i-1]);
do
j--;
while(A[j]==A[j+1]);
count+=(i-cl)*(cr-j);
}
}
else if(s<sum)
i++;
else
j--;
}
return count;
}
};
方法二:哈希表法
class FindPair {
public:
int countPairs(vector<int> A, int n, int sum) {
// write code here
map<int,int> record;
for(int i=0;i<n;i++)
record[A[i]]++;
int result=0;
map<int,int>::iterator it;
map<int,int>::iterator find;
for(it=record.begin();it!=record.end();it++){
find=record.find(sum-it->first);
if(find!=it&&find!=record.end()){
result+=(it->second*find->second);
it->second=find->second=0;
}
else if(find!=record.end()){
result+=(find->second)*(find->second-1)/2;
find->second=0;
}
}
return result;
}
};