51nod 1267 4个数和为0(二分查找)好题!!!

来源:互联网 发布:手机卫士源码 编辑:程序博客网 时间:2024/06/05 05:10


给出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-11-524
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;}







1 0
原创粉丝点击