给出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
将两两相加的数组存起来,再进行二分查找使得sum[l]+sum[r]==0, 并且坐标不重叠!!!
#include<bits/stdc++.h>
using namespace std;
const int maxn=1000+10;
int num[maxn];
struct node
{
int x;
int y;
int sum;
} a[maxn*maxn];
bool cmp(node a,node b)
{
if(a.sum<b.sum) return true;
return false;
}
int main()
{
int n,i,j,ok,k;
ok=0;
cin>>n;
k=0;
for(i=1;i<=n;i++) {
cin>>num[i];
for(j=1;j<i;j++) {
a[k].x=i;
a[k].y=j;
a[k++].sum=num[i]+num[j];
}
}
sort(a,a+k,cmp);
i=0;j=k-1;
while(i<k) {
if(a[i].sum+a[j].sum==0) {
if(a[i].x!=a[j].x && a[i].x!=a[j].y && a[i].y!=a[j].x && a[i].y!=a[j].y) {
ok=1;
break;
}
if(a[i].sum==a[i+1].sum) i++;
else if(a[j].sum==a[j-1].sum) j--;
else {
i++;
j--;
}
}
else if(a[i].sum+a[j].sum<0) i++;
else j--;
}
if(ok) printf("Yes\n");
else printf("No\n");
return 0;
}