4个数和为0(二分)

4个数和为0(二分)


给出N个整数，你来判断一下是否能够选出4个数，他们的和为0，可以则输出"Yes"，否则输出"No"。
Input
第1行，1个数N，N为数组的长度(4 <= N <= 1000) 
第2 - N + 1行：Aii（-10^9 <= Aii <= 10^9)
Output
如果可以选出4个数，使得他们的和为0，则输出"Yes"，否则输出"No"。
Sample Input
5
-1
1
-5
2
4
Sample Output
Yes

#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
//#include<mm_malloc.h>
#include<map>
#define MT(a,b) memset(a,b,sizeof(a))
#define ll long long
using namespace std;
const int maxn = 1e3+1;
int  a[maxn];
struct dd{
    int first,second;
    int num;
}v[2*maxn*maxn];
map<int ,dd>mp;
bool flag=false;
bool com(dd a,dd b)
{
    return a.num<b.num;
}
int main()
{
    int n;cin>>n;
    for(int i=0;i<n;i++)cin>>a[i];
    int l=-1;
    for(int i=0;i<n;i++)
    {
        for(int j=i+1;j<n;j++)
        {
            v[++l].num=a[i]+a[j];
            v[l].first=i;
            v[l].second=j;
        }
    }//数组两两相加v[i]记录大小，first，second记录两个数的下标
    sort(v,v+l,com);//排序
    for(int i=0;i<=l;i++)
    {
        int f=0,r=l,ans=(f+l)/2;
        int x=-v[i].num;//记录的相反数x
        while(f<=r)
        {
            int mid=(f+r)/2;
            if(x>=v[mid].num)ans=mid,f=mid+1;
            else r=mid-1;
        }//查找x的下标
        if(x==v[ans].num)
        {
            f=r=ans;
            while(x==v[f-1].num)f--;
            while(x==v[r+1].num)r++;//计算v[i]等于x的最小和最大下标
            for(int j=f;j<=r;j++)
            {
                if(v[i].first!=v[j].first&&v[i].second!=v[j].first&&v[i].first!=v[j].second&&v[i].second!=v[j].second)
                {//判断四个数是否重复
                    cout<<"Yes"<<endl;
                    return 0;
                }
            }
        }
    }
    cout<<"No"<<endl;
    return 0;
}