基准时间限制:1 秒 空间限制:131072 KB 分值: 20
难度:3级算法题
给出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
现有小到大排序,然后在事前两个数相加,然后依次对后面的数进行二分,知道找到满足条件的解
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
using namespace std;
const int MAXN=10005;
int a[MAXN];
int main(){
int n;
while(~scanf("%d",&n)){
for(int i=0;i<n;i++){
cin>>a[i];
}
sort(a,a+n);
bool flag=false;
for(int i=0;i<n;i++){
if(flag) break;
for(int j=i+1;j<n;j++){
if(flag) break;
//int sum=0;
int left=j+1,right=n-1;
while(left<right){
int sum=a[i]+a[j]+a[left]+a[right];
if(sum==0){
flag=true;
break;
}
else if(sum<0) left++;
else if(sum>0) right--;
}
}
}
if(flag) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
return 0;
}