4个数和为0(二分)
给出N个整数,你来判断一下是否能够选出4个数,他们的和为0,可以则输出"Yes",否则输出"No"。
Input
第1行,1个数N,N为数组的长度(4 <= N <= 1000)
第2 - N + 1行:Aii(-10^9 <= Aii <= 10^9)
Output
如果可以选出4个数,使得他们的和为0,则输出"Yes",否则输出"No"。
Sample Input
5
-1
1
-5
2
4
Sample Output
Yes
给出N个整数,你来判断一下是否能够选出4个数,他们的和为0,可以则输出"Yes",否则输出"No"。
Input
第1行,1个数N,N为数组的长度(4 <= N <= 1000)
第2 - N + 1行:Aii(-10^9 <= Aii <= 10^9)
Output
如果可以选出4个数,使得他们的和为0,则输出"Yes",否则输出"No"。
Sample Input
5
-1
1
-5
2
4
Sample Output
Yes
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cstdio>
//#include<mm_malloc.h>
#include<map>
#define MT(a,b) memset(a,b,sizeof(a))
#define ll long long
using namespace std;
const int maxn = 1e3+1;
int a[maxn];
struct dd{
int first,second;
int num;
}v[2*maxn*maxn];
map<int ,dd>mp;
bool flag=false;
bool com(dd a,dd b)
{
return a.num<b.num;
}
int main()
{
int n;cin>>n;
for(int i=0;i<n;i++)cin>>a[i];
int l=-1;
for(int i=0;i<n;i++)
{
for(int j=i+1;j<n;j++)
{
v[++l].num=a[i]+a[j];
v[l].first=i;
v[l].second=j;
}
}//数组两两相加v[i]记录大小,first,second记录两个数的下标
sort(v,v+l,com);//排序
for(int i=0;i<=l;i++)
{
int f=0,r=l,ans=(f+l)/2;
int x=-v[i].num;//记录的相反数x
while(f<=r)
{
int mid=(f+r)/2;
if(x>=v[mid].num)ans=mid,f=mid+1;
else r=mid-1;
}//查找x的下标
if(x==v[ans].num)
{
f=r=ans;
while(x==v[f-1].num)f--;
while(x==v[r+1].num)r++;//计算v[i]等于x的最小和最大下标
for(int j=f;j<=r;j++)
{
if(v[i].first!=v[j].first&&v[i].second!=v[j].first&&v[i].first!=v[j].second&&v[i].second!=v[j].second)
{//判断四个数是否重复
cout<<"Yes"<<endl;
return 0;
}
}
}
}
cout<<"No"<<endl;
return 0;
}