满足条件的整数对个数

题目描述

请设计一个高效算法，找出数组中两数之和为指定值的所有整数对。

给定一个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;
    }
};