[51nod] 1267 4个数和为0

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

Analysis分析

　我从Hash的题目列表点进来，，，结果发现一堆人写排序

　= =

　如果枚举的话是 n⁴ 那么我们枚举两个数的和，然后用Hash存起来

　然后再次枚举数和，检测其相反数是否存在（当然这里需要再加一个所有四个元素下标不重复的判定，如果是 1 -1 0 这种那么就会出事）

　Emmm加补丁的Hash

 

Code代码

#include<stdio.h>
#include<iostream>
#define ULL unsigned long long
#define maxn 1000007
using namespace std;

// 上面maxn那里我一开始设的是100007，结果TLE qwq

struct data{ ULL val,pos; }sum[maxn];
int m,n,arr[maxn];

struct edge{ ULL val,pos; int from; }e[maxn*4];
int tot,first[maxn];

bool noequ(ULL A,ULL B){
    return (A/10000 != B/10000) && (A%10000 != B%10000);
}

bool check(ULL val,ULL pos){
    int w = val%maxn;
    for(int i = first[w];i;i = e[i].from){
        if(e[i].val == val && noequ(e[i].pos,pos)) return true;
    }return false;
}

void insert(int pos1,int pos2,int suuum){
    if(pos1 > pos2) swap(pos1,pos2);
    ULL val = (ULL)(suuum+1e9);
    ULL kcode = pos1*10000+pos2;
    
    if(check(val,kcode)) return;
    
    int w = val%maxn;
    
    tot++; e[tot].from = first[w];
    e[tot].val = val; e[tot].pos = kcode;
    first[w] = tot;
    sum[++m].val = val;
    sum[m].pos = kcode;
}

int main(){
    scanf("%d",&n);
    
    for(int i = 1;i <= n;i++)
        scanf("%d",&arr[i]);
    
    for(int i = 1;i <= n;i++)
        for(int j = i+1;j <= n;j++)
            insert(i,j,arr[i]+arr[j]);
    
//    for(int i = 1;i <= m;i++) printf("%I64u %I64u
",sum[i].val,sum[i].pos);
//    
//    cout << "Hash:" << endl;
//    
//    for(int i = 1;i <= tot;i++) printf("%I64u %I64u
",e[i].val,e[i].pos);
    
    for(int i = 1;i <= m;i++){
        if(check(2e9-sum[i].val,sum[i].pos)){
            cout << "Yes";
            return 0;
        }
    }
    
    cout << "No";
    
    return 0;
}