51nod 1267 4个数和为0

【题解】

　　先把数字两两组合，问题转化为求a+b=0. 这个问题显然可以排序后two pointer做。需要注意选出来的两个数必须由4个不同位置的数组成，不能重复。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define rg register
#define N 1000010
using namespace std;
int n,tot,a[N],b[N],pos[N][2];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
int main(){
    n=read();
    for(rg int i=1;i<=n;i++) b[i]=read();
    for(rg int i=1;i<n;i++)
        for(rg int j=i+1;j<=n;j++)
        a[++tot]=b[i]+b[j],pos[tot][0]=i,pos[tot][1]=j;
    sort(a+1,a+1+tot);
    int l=1,r=tot;
    while(l<r){
        while(a[l]+a[r]>0) r--;
        if(a[l]+a[r]==0){
            for(rg int i=r;i>=l;i--)if(a[l]+a[r]==0){
                if(pos[l][0]==pos[r][0]||pos[l][1]==pos[r][0]||pos[l][0]==pos[r][1]||pos[l][0]==pos[r][1]) continue;
                puts("YES");
                return 0;
            }
            else break;
        }
        l++;
    }
    puts("NO");
    return 0;
}