Problem Killer
这一题简单的模拟题,大致题意:有一串数,取其中连续的一串数,使其为等比或等差数列,求最长的长度。
因为必须是连着的,所以从头扫到尾,如果断了,从断点处重新开始,O(n)的时间复杂度
代码附上
#include<iostream>
#include<cstdio>
using namespace std;
int T,i,n,dc,dbmax,dcmax,dbm,dcm,a[1000012];
double db;
int max(int a,int b)
{
if (a>b) return a; else return b;
}
int main()
{
cin>>T;
for (int I=1;I<=T;I++)
{
scanf("%d",&n);
for (i=1;i<=n;i++) scanf("%d",&a[i]);
if (n<=1) cout<<n<<endl;
else
{
i=3;
dc=a[2]-a[1]; dbmax=2; dbm=2;
db=1.0*a[2]/a[1]; dcmax=2; dcm=2;
while (i<=n)
{
if (a[i]-a[i-1]==dc) dcm++;
else
{
if (dcm>dcmax) dcmax=dcm;
dcm=2; dc=a[i]-a[i-1];
}
if (1.0*a[i]/a[i-1]==db) dbm++;
else
{
if (dbm>dbmax) dbmax=dbm;
dbm=2; db=1.0*a[i]/a[i-1];
}
i++;
}
if (dcm>dcmax) dcmax=dcm;
if (dbm>dbmax) dbmax=dbm;
cout<<max(dbmax,dcmax)<<endl;
}
}
}