1,题意
给出n个数,让你利用这些书构造出一个最长的生成序列.
要求序列是一个等差数列或等比数列或等幂数列
2,解答
3,代码
给出n个数,让你利用这些书构造出一个最长的生成序列.
要求序列是一个等差数列或等比数列或等幂数列
2,解答
3,代码
#include <iostream>
#include <map>
using namespace std;
typedef long long LL;
map<LL,int> m; //保存元素
LL d[100]; //输入序列
int cnt;//测试数
int n;//元素个数
int ans;//记录序列长度
bool in_range(LL a,int k) //检查a^k是否在[-2^31 2^31)之间
{
LL ret=1;
while(k--)
{
ret*=a;
if(ret>0x7fffffff||-ret>0x8fffffff)
return false;
}
return true;
}
LL power(LL a,int k) //求a^k
{
LL ret=1;
while(k--) ret*=a;
return ret;
}
void findNum(LL a,int k,int st)
{
int l=0; //记录当前长度
switch(st)
{
case 1: //寻找等差数列
for(LL i=a;;i+=k)
{
//这个元素存在,序列长度增加
if(m[i]) m[i]--,l++;
else break;
}
if(l>ans) ans=l;
for(int i=a;l;i+=k)
{
m[i]++;
l--;
}
break;
case 2: //寻找等比数列
for(LL i=a;;i*=k)
{
//这个元素存在,序列长度增加
if(m[i]) m[i]--,l++;
else break;
}
if(l>ans) ans=l;
for(int i=a;l;i*=k)
{
m[i]++;
l--;
}
break;
case 3: //寻找等幂序列
//注意这里的处理
for(LL i=a;;i=power(i,k))
{
//这个元素存在,序列长度增加
if(m[i]) m[i]--,l++;
else break;
if(!in_range(i,k)) break;
}
if(l>ans) ans=l;
for(int i=a;l;i=power(i,k))
{
m[i]++;
l--;
}
break;
}
}
int main()
{
freopen("5.18.in","r",stdin);
cin>>cnt;
while(cnt--)
{
cin>>n;
ans=1;//长度至少为1
m.clear();
for(int i=0;i<n;i++)
{
cin>>d[i];
m[d[i]]++;
}
for(int i=0;i<n;i++)
for(int j=0;j<n;j++) //取出任意两个数
{
if(i!=j)
{
findNum(d[i],d[j]-d[i],1); //找出最长的等差数列
//找等比数列,保证d[i]!=0,且可以整除
if( d[i]&&d[j]%d[i]==0 )
findNum(d[i],d[j]/d[i],2);
//找等幂序列,分好几种情况
switch(d[i])
{
case 0:
if(d[j]==0) //只能取0
findNum(0,1,3);
break;
case 1:
if(d[j]==1) //只能取1
findNum(1,1,3);
break;
case -1:
if(d[j]==-1) //k为奇数
findNum(-1,1,3);
if(d[j]==1) //k为偶数
findNum(-1,0,3);
break;
default:
for(int k=0;in_range(d[i],k);k++)
{
LL f=power(d[i],k);
if(f>abs(d[j])) break; //该元素不存在
else if(f==d[j]) //找到了当前的k
{
findNum(d[i],k,3);
break;
}
}
}
}
}
cout<<ans<<endl;
}
return 0;
}