给出N个整数,你来判断一下是否能够选出4个数,他们的和为0,可以则输出”Yes”,否则输出”No”。
Input
第1行,1个数N,N为数组的长度(4 <= N <= 1000)
第2 - N + 1行:Ai(-10^9 <= Ai <= 10^9)
Output
如果可以选出4个数,使得他们的和为0,则输出”Yes”,否则输出”No”。
Sample Input
5
-1
1
-5
2
4
Sample Output
Yes
思路:
写四重for循环暴力搜索显然会超时。故本题先将序列由小到大排序,依次求前两个数的和,再使用二分在序列中查找后两个数与前两个数的和为0。
代码:
#include <stdio.h>
#include <stdlib.h>
long long a[1010],sum;
void quicksort(int left,int right);
int main()
{
int i,j,n,l,r;
scanf("%d",&n);
for(i=1;i<=n;i++)
scanf("%lld",&a[i]);
quicksort(1,n);
for(i=1;i<n;i++)
for(j=i+1;j<=n;j++)
{
sum=-a[i]-a[j]; //前两个数的和
l=j+1;
r=n;
while(l<r) //二分找后两个数
{
if(a[l]+a[r]==sum)
{
printf("Yes\n");
return 0;
}
else if(a[l]+a[r]>sum)
r--;
else
l++;
}
}
printf("No\n");
return 0;
}
void quicksort(int left,int right)
{
int i,j,key;
if(left>right)
return;
i=left;
j=right;
key=a[left];
while(i!=j)
{
while(a[j]>=key&&i<j)
j--;
a[i]=a[j];
while(a[i]<=key&&i<j)
i++;
a[j]=a[i];
}
a[i]=key;
quicksort(left,i-1);
quicksort(i+1,right);
}