EOJ2530 数字之和

http://acm.cs.ecnu.edu.cn/problem.php?problemid=2530

题意：求一个数组中是否存在三个数的和为0。n=1000，朴素的n^3会超时，故用二分搜索，即是把正负数存入不同的两个数组，在一个数组中任取两个，求和，然后在另一个数组中利用二分查找其相反数是否存在， 复杂度将降为n^2logn 级别

#include<map>
#include<cmath>
#include<cctype>
#include<cstdio>
#include<string>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
bool b_search(int key,int *num,int start,int end){
    while(start<=end){
        int m=(start+end)/2;
        if(num[m]>key)
            end=m-1;
        else if(num[m]<key)
            start=m+1;
        else return true;
    }
    return false;
}    
int main(){
    int T;
    cin>>T;
    while(T--){
        int n,i=0,j=0;
        int pos[1005],neg[1005];
        cin>>n;
        while(n--){
            int x;
            scanf("%d",&x);
            if(x>=0)pos[i++]=x;           //存入不同数组
            else neg[j++]=x;
        }
        sort(pos,pos+i);
        sort(neg,neg+j);
        bool f=0;
        for(int k=0;k<i;k++)
            for(int h=k+1;h<i;h++){                //任取两个，然后二分查找
                int temp=pos[k]+pos[h];
                if(b_search(-1*temp,neg,0,j-1)){f=1;break;}
            }
        if(!f){
            for(int k=0;k<j;k++)
                for(int h=k+1;h<j;h++){
                    int temp=neg[k]+neg[h];
                    if(b_search(-1*temp,pos,0,i-1)){f=1;break;}
                }
        }
        if(f)cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }
    return 0;
}

 我又来hash了。。。

#include <iostream>
#include <stdio.h>
#include <string>
#include <algorithm>
#include <string.h>
#include <stdlib.h>

using namespace std;
short hash[300005];

int main()
{
    //freopen("testin.txt", "r", stdin);
    //freopen("testout.txt", "w", stdout);
    
    int t;
    cin >> t;
    while(t--)
    {
        int n, na, nb, ia=0, ib=0;
        int a[1005], b[1005];
        cin >> n;
        while(n--)
        {
            int x;
            scanf("%d", &x);
            if(x >= 0)
                a[ia++] = x;
            else
                b[ib++] = x;
        }
        na = ia, nb = ib;
        memset(hash, 0, sizeof(hash));
        for(int i=0; i<na-1; i++)
            for(int j=i+1; j<na; j++)
                hash[a[i]+a[j]] ++;
        int isyes = 0;
        for(int i=0; i<nb; i++)
            if(hash[-1*b[i]])
            {
                isyes = 1; 
                break;
            }
        if(isyes)
            cout << "YES" << endl;
        else
            cout << "NO" << endl; 
    }
    
    
    return 0;
}