HDU 5328 Problem Killer(简单DP)

来源:互联网 发布:网络流行词汇 编辑:程序博客网 时间:2024/05/11 01:59

大意:

输出其中最长的等差(AP)或等比(GP)数列的长度。

数列长度为1或2时一定是等比或等差数列。

如果a,b,c是AP或GP,b,c,d是AP或GP那么a,b,c,d是AP或GP。

因此有状态转移方程。

#include <iostream>#include <stdio.h>#include <math.h>#include <memory.h>#define eps 1e-6;using namespace std;int a[1000005];int max1=2;int ap[1000005];int gp[1000005];void solve(int n){    ap[0]=2;    ap[1]=2;    gp[0]=2;    gp[1]=2;    for(int i=2; i<n; i++)    {        ap[i]=2;        gp[i]=2;        if(a[i-1]*2==a[i-2]+a[i])        {            ap[i]=ap[i-1]+1;            if(ap[i]>max1)                max1=ap[i];        }        if(fabs(a[i]*1.0/a[i-1]-a[i-1]*1.0/a[i-2])<1e-6)        {            gp[i]=gp[i-1]+1;            if(gp[i]>max1)                max1=gp[i];        }    }}int main(){    int n,T;    memset(a,0,sizeof(a));    scanf("%d",&T);    while(T--)    {        scanf("%d",&n);        for(int i=0; i<n; i++)            scanf("%d",&a[i]);        if(n==1||n==2)        {            printf("%d\n",n);        }        else        {            max1=2;            solve(n);            printf("%d\n",max1);        }    }    return 0;}


0 0
原创粉丝点击