题目链接:https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1267点击打开链接
基准时间限制: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 <bits/stdc++.h>
using namespace std;
struct xjy
{
int sum;
int num1,num2;
int hashh;
bool operator < (const xjy &r)const
{
return sum<r.sum;
}
};
struct xxjy
{
int num1,num2;
};
vector<int > s;
vector<xjy > ssum;
vector<xxjy>judge[1111111];
vector<int > ffind;
map<int ,int > mmap;
int main()
{
int n;
scanf("%d",&n);
for(int i=0;i<n;i++)
{
int mid;
scanf("%d",&mid);
s.push_back(mid);
}
for(int i=0;i<n;i++)
for(int j=i+1;j<n;j++)
{
xjy mid;
mid.sum=s[i]+s[j];
mid.num1=i;
mid.num2=j;
ssum.push_back(mid);
}
sort(ssum.begin(),ssum.end());
int tot=ssum.size();
int cnt=1;
for(int i=0;i<tot;i++)
{
if(mmap[ssum[i].sum])
{
xxjy mid;
mid.num1=ssum[i].num1;
mid.num2=ssum[i].num2;
judge[mmap[ssum[i].sum]].push_back(mid);
}
else
{
mmap[ssum[i].sum]=cnt++;
xxjy mid;
mid.num1=ssum[i].num1;
mid.num2=ssum[i].num2;
judge[mmap[ssum[i].sum]].push_back(mid);
ffind.push_back(ssum[i].sum);
}
}
int ans=0;
for(int i=0;i<n;i++)
for(int j=i+1;j<n;j++)
{
int midsum=s[i]+s[j];
midsum=0-midsum;
int l=0;int r=ffind.size()-1;
while(l<=r)
{
int mid=(l+r)>>1;
if(ffind[mid]<=midsum)
{
ans=mid;
l=mid+1;
}
else
r=mid-1;
}
if(ffind[ans]!=midsum)
continue;
int midtot=judge[mmap[ffind[ans]]].size();
for(int k=0;k<midtot;k++)
{
if(judge[mmap[ffind[ans]]][k].num1==i||judge[mmap[ffind[ans]]][k].num1==j||judge[mmap[ffind[ans]]][k].num2==i||judge[mmap[ffind[ans]]][k].num2==j)
continue;
else
{
printf("Yes\n");
return 0;
}
}
}
printf("No\n");
}