codeforces255C.Almost Arithmetical Progression

frog&7發表於2020-11-04

題意:求出最長的i j i j i…j(i) 序列的長度
思路:先把各個數字離散化成1-n的形式,可以減少記憶體的使用,方便動態規劃處理,然後用 dp[i][j] 表示最後兩位為i j 的最大長度

#pragma GCC optimize("Ofast")
#pragma GCC target("avx,avx2,fma")
#pragma GCC optimization ("unroll-loops")
#include<bits/stdc++.h>
using namespace std;
int dp[4005][4005];
const int sz=1e6+5;
int order[sz];
int main()
{
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int ans=0;
    int n;
    cin>>n;
    vector<int> b(n+1);
    set<int> st;
    for(int i=1;i<=n;i++){
            cin>>b[i];
            st.insert(b[i]);}
    if(n<=2)cout<<n<<endl;
    else{
    int k=1;
    for(auto it:st){
        order[it]=k;
        k++;
    }
    for(int i=1;i<=n;i++){
        b[i]=order[b[i]];
    }
    for(int i=1;i<=n;i++){
        bool flag=0; 
        for(int j=i-1;j>=0;j--){
            if(b[i]==b[j]){
                if(flag)continue;
                flag=1;
                dp[b[i]][b[j]]++;
                ans=max(ans,dp[b[i]][b[j]]);
            }
            else{
                dp[b[i]][b[j]]=max(dp[b[i]][b[j]],dp[b[j]][b[i]]+1);
                ans=max(ans,dp[b[i]][b[j]]);
            }
        }
    }
    cout<<ans+1<<endl;

}
}