4个数和为0 51Nod - 1267
题目描述:
给出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
先排序,自左向右遍历枚举两个值,求和sum,然后从剩下的区间内从两边逼近直到找到和为-1*sum的另外两个值;
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long
using namespace std;
ll a[1050];
int main(){
int n,p=0;
cin>>n;
for(int i=0;i<n;i++)
scanf("%lld",&a[i]);
sort(a,a+n);
for(int i=0;i<n-3;i++){
for(int j=i+1;j<n-2;j++){
int l=j+1,r=n-1,sum=-1*(a[i]+a[j]);
while(l<r){
if(a[l]+a[r]<sum)
l++;
else if(a[l]+a[r]>sum)
r--;
else {p=1;break;}
}
if(p) break;
}
if(p) break;
}
if(p) cout<<"Yes"<<endl;
else cout<<"No"<<endl;
return 0;
}