Codeforces 934C - A Twisty Movement

934C - A Twisty Movement

思路：dp

很容易想到要预处理出1的前缀和pre[i]和2的后缀和suf[i]

然后枚举区间，对于每个区间如果能求出最长递减序列的长度，那么就能更新答案了

这个用dp求

状态：

dp[i][j][0]表示i--j区间以2结尾的最长递减序列长度，很明显这个序列全为2，所以也就是i--j区间2的个数

dp[i][j][1]表示i--j区间以1结尾的最长递减序列长度

状态转移：

dp[i][j][0]=dp[i][j-1][0]+(a[j]==2)
dp[i][j][1]=max(dp[i][j-1][0],dp[i][j-1][1])+(a[j]==1)

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=2e3+;
int a[N];
int pre[N];
int suf[N];
int dp[N][N][];
int main(){
    ios::sync_with_stdio(false);
    cin.tie();
    int n;
    cin>>n;
    for(int i=;i<=n;i++)cin>>a[i];
    for(int i=;i<=n;i++)pre[i]=pre[i-]+(a[i]==);
    for(int i=n;i>=;i--)suf[i]=suf[i+]+(a[i]==);
    int ans=;
    for(int i=;i<=n;i++){
        for(int j=i;j<=n;j++){
            dp[i][j][]=dp[i][j-][]+(a[j]==);
            dp[i][j][]=max(dp[i][j-][],dp[i][j-][])+(a[j]==);
            ans=max(ans,pre[i-]+suf[j+]+dp[i][j][]);
            ans=max(ans,pre[i-]+suf[j+]+dp[i][j][]);
        }
    }
    cout<<ans<<endl;
    return ;
}