51nod 1267 二分

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

1267 4个数和为0

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题

收藏

关注

给出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<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
#include<ctime>
#include<functional>
#include<algorithm>
using namespace std;
#define LL long long
int A[1005];
struct node 
{ 
    LL w; 
    int  a, b; 
    node() {};
    node(LL s, int a, int b):w(s), a(a), b(b) {};
    bool operator<(const node& tmp)const {
        return w < tmp.w;
    }
}P[1000005];
int main()
{
    int N, i, j, k,p=0;
    cin >> N;
    for (i = 1;i <= N;++i)
    {
        scanf("%d", &A[i]);
        for (j = 1;j < i;++j)
        {
            p++;
            P[p].w = (LL)A[i] + A[j];
            P[p].a = j;
            P[p].b = i;
        }
    }
    P[p + 1].w = 1e18;
    bool ok = 0;
    sort(P + 1, P + 1 + p);
    for (i = 1;i <= p;++i)
    {
        int k = lower_bound(P+1,P+2+p,node(-P[i].w,0,0))-P;
        if (P[k].w != -P[i].w) continue;
        for (j = k;P[j].w == -P[i].w&&j <= p;j++) {
            if (P[i].a!=P[j].a&&P[i].a!=P[j].b&&P[i].b!=P[j].a&&P[i].b!=P[j].b) { ok = 1;break; }
        }
        if (ok)break;
    }
    ok ? puts("Yes") : puts("No");
    //system("pause");
    return 0;
}