51Nod - 1267 4个数和为0

 

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

Input 第1行，1个数N，N为数组的长度(4 <= N <= 1000)
第2 - N + 1行：A ii（-10^9 <= A ii <= 10^9) Output 如果可以选出4个数，使得他们的和为0，则输出"Yes"，否则输出"No"。 Sample Input

5
-1
1
-5
2
4

Sample Output

Yes

思路：你可以两两相加，从这一新序列中挑选A、B两个数，（需要注意的是，你挑选的这两个数A、B有可能是由三个数相加而得，所以当A+B==0时，你需要判断A、B分别是由谁而来，若它们来自不同的数，则输出“Yes”）然后利用二分，for循环遍历一个数a，二分找另一个数b。（当你查找的时候很可能遇到很多与a相加等于0，这种情况无需多考虑，因为你遍历了整个序列，只有当你做标记的x==-1时，才输出“No”）

若a+b==0   a==1+2，b==3+4，  则（1、2）+（3、4）==0   （1、3）+（2、4）==0  （1、4）+（2、3）==0

代码参考：

#include <iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<cstring>
#define max(a,b) a>b?a:b
#define ll long long
#define inf 0x3f3f3f3f

using namespace std;
ll a[1010];
struct node{
    int x;
    int y;
    ll z;
};
node s[600050];
bool cmp(node xx,node yy)
{
    return xx.z<yy.z;
}
int main()
{
    ll n;
    while(~scanf("%lld",&n))
    {
        memset(a,inf,sizeof(a));
        ll p=0;
        for(ll i=0;i<n;i++)
            scanf("%lld",&a[i]);
        for(ll i=0;i<n;i++)
        {
            for(ll j=i+1;j<n;j++)
            {
                s[p].x=i;
                s[p].y=j;
                s[p].z=a[i]+a[j];
                p++;
            }
        }
        sort(s,s+p,cmp);
        ll k=-1,l=-1;
        for(ll i=0;i<p;i++)
        {
            ll L=0,R=p-1,mid=(L+R)/2;
            while(L<R)
            {
                if(s[i].z+s[mid].z<0)
                    L=mid+1;
                else if(s[i].z+s[mid].z>0)
                    R=mid;
                else
                    break;
                mid=(L+R)/2;
            }
            if(s[i].z+s[mid].z==0)
            {
                if(s[i].x!=s[mid].x && s[i].x!=s[mid].y &&
                   s[i].y!=s[mid].x && s[i].y!=s[mid].y)
                {
                    k=i;
                    printf("Yes\n");break;
                }
            }
        }
        if(k==-1)
            printf("No\n");
    }
    return 0;
}