BZOJ 3357: [Usaco2004]等差数列 动态规划

Code:

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 2003
using namespace std;
map<int,int>k;
int arr[maxn],f[maxn][maxn];
int main() {
    // setIO("input");
    int n,ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n;++i) scanf("%d",&arr[i]);
    if(n==1) {
        printf("1\n");
        return 0;
    }
    for(int i=1;i<n;++i) f[i][n]=2;
    k[arr[n]]=n;
    for(int j=n-1;j>1;--j) {
        for(int i=1;i<j;++i) {
            f[i][j]=2;
            int to=arr[j]+arr[j]-arr[i];
            if(k[to]) {
                f[i][j]=max(f[i][j], f[j][k[to]] + 1);
            }
            ans=max(f[i][j], ans);
        }
        k[arr[j]]=j;
    }
    printf("%d\n",ans);
    return 0;
}