51nod——T1267 4个数和为0

https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1267

题目描述

给出N个整数，你来判断一下是否能够选出4个数，他们的和为0，可以则输出"Yes"，否则输出"No"。

Input

第1行，1个数N，N为数组的长度(4 <= N <= 1000)
第2 - N + 1行：A[i]（-10^9 <= A[i] <= 10^9)

Output

如果可以选出4个数，使得他们的和为0，则输出"Yes"，否则输出"No"。

Input示例

5
-1
1
-5
2
4

Output示例

Yes

#include <algorithm>
#include <iostream>

using namespace std;

int n,a[1050],cnt;
int if_ok;

struct node_num
{
    int l,r,tot;
}num[1000*1000+5];

bool cmp(node_num a,node_num b)
{
    return a.tot<b.tot;
}

int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>a[i]; sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
        {
            num[++cnt].l=a[i];
            num[cnt].r=a[j];
            num[cnt].tot=a[i]+a[j];
        }    
    sort(num+1,num+cnt+1,cmp);
    int l=1,r=cnt;
    while(l<r)
    {
        int sum=num[l].tot+num[r].tot;
        if(sum==0&&num[l].l!=num[r].l&&num[l].l!=num[r].r&&num[l].r!=num[r].l&&num[l].r!=num[r].r)
        { if_ok=1; break; }
        if(sum<0) l++;
        else r--;
/*        int midd=(l+r)/2;
        if(num[l].tot+num[midd].tot<0) l=midd+1;
        if(num[r].tot+num[midd].tot>0) r=midd-1;
*/    }
    if(if_ok) cout<<"Yes";
    else cout<<"No";
    return 0;
}