bzoj3357 [Usaco2004]等差数列 dp+map

Description

---

约翰发现奶牛经常排成等差数列的号码．他看到五头牛排成这样的序号：“1，4，3，5，7”
很容易看出“1，3，5，7”是等差数列．
给出N(1≤N≤2000)数字AI..AN(O≤Ai≤10^9)，找出最长的等差数列，输出长度．

Solution

---

容易想到dp，dp的时候注意到决定等差数列的是任意两数的差和首项，那么设f[i][a[j]]表示当前以a[j]、a[i]作为结尾的等差数列长度。由性质易得所求a[k]=a[j]*2-a[i]
之所以这样做是因为可以用map省去哈希操作，学习了
注意n=1的时候答案为1

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <map>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
const int N=2025;
std:: map<int,int> f[N];
int a[N];
int main(void) {
    int n; scanf("%d",&n);
    rep(i,1,n) scanf("%d",&a[i]);
    int ans=0;
    rep(i,1,n) {
        rep(j,1,i-1) {
            f[i][a[j]]=std:: max(f[i][a[j]],2);
            f[i][a[j]]=std:: max(f[i][a[j]],f[j][a[j]*2-a[i]]+1);
            ans=std:: max(ans,f[i][a[j]]);
        }
    }
    if (n==1) ans=1;
    printf("%d\n", ans);
    return 0;
}