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[];
 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[];
 int main()
 {
     int N, i, j, k,p=;
     cin >> N;
     for (i = ;i <= N;++i)
     {
         scanf("%d", &A[i]);
         for (j = ;j < i;++j)
         {
             p++;
             P[p].w = (LL)A[i] + A[j];
             P[p].a = j;
             P[p].b = i;
         }
     }
     P[p + ].w = 1e18;
     bool ok = ;
     sort(P + , P +  + p);
     for (i = ;i <= p;++i)
     {
         int k = lower_bound(P+,P++p,node(-P[i].w,,))-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 = ;break; }
         }
         if (ok)break;
     }
     ok ? puts("Yes") : puts("No");
     //system("pause");
     return ;
 }