4个数和为0
51Nod - 1267
给出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"。 Sample Input
5 -1 1 -5 2 4Sample Output
Yes
code:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1011;
int n;
struct node{
ll s;
int i,j;
}sum[maxn*maxn];
ll a[maxn];
int cnt = 0;
int id = -1;
bool cmp(node a,node b){
return a.s < b.s;
}
int F(ll x){
int l = 0,r = cnt-1;
while(l <= r){
int mid = (l + r) >> 1;
if(sum[mid].s == x){
id = mid;
return 1;
}
else if(sum[mid].s < x)
l = mid + 1;
else
r = mid - 1;
}
return 0;
}
int main(){
scanf("%d",&n);
for(int i = 0; i < n; i++){
scanf("%lld",&a[i]);
}
for(int i = 0; i < n; i++){
for(int j = i+1; j < n; j++){
sum[cnt].s = a[i] + a[j];
sum[cnt].i = i;
sum[cnt++].j = j;
}
}
sort(sum,sum+cnt,cmp);
int flag = 0;
for(int i = 0; i < cnt; i++){
id = -1;
int k = F(-sum[i].s);
if(id == -1) continue;//如果没找到跳过
if(sum[i].i == sum[id].i || sum[i].i == sum[id].j || sum[i].j == sum[id].i || sum[i].j == sum[id].j)//确保没有重复的数字
continue;
if(k){
flag = 1;
break;
}
}
if(!flag) printf("No\n");
else printf("Yes\n");
return 0;
}