1267 4个数和为0
基准时间限制: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
题解:
虽然是参考了别人的思路。紫书上也有一道类似的题,
用到在51nod里做过的一道数组中和为k的数对,一样的思想。
二分+排序
具体看代码
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int maxn = 1e3+10;
LL num[maxn];
int main()
{
int n;
cin>>n;
for(int i=0;i<n;i++)
cin>>num[i];
sort(num,num+n);
for(int i=0;i<n;i++)
for(int j=i+1;j<n;j++)
{
LL sum=-num[i]-num[j];
int left=j+1,right=n-1;
while(left<right)
{
if(num[left]+num[right]==sum)
{
cout<<"Yes"<<endl;
return 0;
}
else if(num[left]+num[right]>sum)
{
right--;
}
else
left++;
}
}
cout<<"No"<<endl;
return 0;
}