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


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










你可能感兴趣的:(51nod 1267 4个数和为0(二分查找)好题!!!)