基准时间限制: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; }