题目大意:给定一个序列,让我们判断是否满足一定的条件,条件如下:
对于任意的i,j,k,1<=i<j<k<=n,都有ai+aj+ak=al,其中1<=l<=n。
解题思路:如果有3个大于0的数字,那最大的三个数相加一定不会出现在原来数列中,同样的,如果存在3个小于0的数字,那么最小的三个数相加也一定不会出现在原数列中,所以大于0的数最多有2个,小于0的数最多有2个,至于0,当0的个数大于3个的时候和有3个0的情况是同样的,所以对于0我们最多只需要保留3个就可以了,这样如果一个数列满足我们前面所提到的所有条件,那这个数列最多就只有7个数字,这时候直接暴力枚举即可。一个思维题竟然卡了1个多小时,哎QAQ
上代码:
#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <map>
#include <vector>
using namespace std;
const int N=2e5+10;
long long a[N];
map<long long ,int>mp;
void solve()
{
int n;
cin>>n;
mp.clear();
//将正数,负数,0分别存放到对应的vector中
vector<long long>zheng;
vector<long long>fu;
vector<long long>zero;
int cnt=0,zcnt=0,fcnt=0,cnt0=0;
for(int i=1;i<=n;i++)
{
long long temp;
cin>>temp;
mp[temp]=1;
if(temp>0)
zheng.push_back(temp) ;
else if(temp<0)
fu.push_back(temp);
else if(temp==0)
{
if(zero.size()>3)
continue;
else
zero.push_back(temp);
}
}
if(zheng.size()>2||fu.size()>2)//正数和负数的数量大于2的情况
{
cout<<"NO"<<endl;
return ;
}
else
{
int flag=1;
for(int i=0;i<zheng.size();i++)
zero.push_back(zheng[i]);
for(int i=0;i<fu.size();i++)
zero.push_back(fu[i]);
//暴力枚举所有情况
for(int i=0;i<zero.size();i++)
for(int j=i+1;j<zero.size();j++)
for(int k=j+1;k<zero.size();k++)
if(!mp[zero[i]+zero[j]+zero[k]])
{
flag=0;
break;
}
if(flag)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
}
int main()
{
int t;
cin>>t;
while(t--)
solve();
return 0;
}