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 ;

 }