三维递推dp Logo Turtle CodeForces - 132C

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
using namespace std;

const int maxn=110;
const int inf = 0x3f3f3f3f;
char s[maxn];
int n;
int dp[110][55][2];   //表示状态 前i位前j个操作后的朝向 (0正1负)
int main(){ 
    scanf("%s",s+1);
    scanf("%d",&n);
    int len = strlen(s+1);
    for(int i=0; i<=len; i++){
        for(int j=0; j<=n; j++){
            dp[i][j][1] = dp[i][j][0] = -inf;
        }
    }
    dp[0][0][1]=dp[0][0][0]=0;     //初始化以及j、k的循环始末要想仔细
                                    //记忆化暴力
    for(int i=1; i<=len; i++){
        for(int j=0; j<=n; j++){       //前j个操作是从前j-k个操作(前i-1位的)递推来的
            for(int k=0; k<=j; k++){   //选择性分配
                if(s[i]==‘T‘){
                    if(k&1){           //多维dp，把i、j用相似的递推思想处理
                        dp[i][j][0] = max(dp[i][j][0], dp[i-1][j-k][0]+1);
                        dp[i][j][1] = max(dp[i][j][1], dp[i-1][j-k][1]-1);
                    }   
                    else{
                        dp[i][j][0] = max(dp[i][j][0], dp[i-1][j-k][1]);
                        dp[i][j][1] = max(dp[i][j][1], dp[i-1][j-k][0]);
                    }
                }
                else{
                    if(k&1){
                        dp[i][j][0] = max(dp[i][j][0], dp[i-1][j-k][1]);
                        dp[i][j][1] = max(dp[i][j][1], dp[i-1][j-k][0]);
                    }
                    else{
                        dp[i][j][0] = max(dp[i][j][0], dp[i-1][j-k][0]+1);
                        dp[i][j][1] = max(dp[i][j][1], dp[i-1][j-k][1]-1);
                    }
                }
            }
        }
    }
    printf("%d\n", max(dp[len][n][0], dp[len][n][1]));
    return 0;
}