[题解]P4302 [SCOI2003] 字符串折叠

思路

区间 DP 好题。定义 $dp_{i,j}$ 表示将 $s_{i \sim j}$ 折叠能获得的最短长度。

那么，依旧是枚举一个中间点 $k$，那么，我们将 $dp_{i,j}$ 分为了 $dp_{i,k}$ 和 $dp_{k + 1,j}$ 两部分。

对于这两部分，可显然，有：$dp_{i,j} = \min(dp_{i,k} + dp_{k + 1,j})$。

如果可以将 $[i,j]$ 区间合并，可以枚举折叠后剩余的字符串的最后一位的下标 $k$，然后判断一下如果是 $k$，能否折叠，如果可以，则有：$dp_{i,j} = \min(dp_{i,i + k - 1} + num_{cnt} + 2)$。

其中，$cnt$ 表示折叠后字符串中的数字，$num_i$ 表示 $i$ 数字的长度，$2$ 表示括号。

递推边界显然，对于所有的 $dp_{i,i} = 1$。

Code

#include <bits/stdc++.h>  
#define re register  
  
using namespace std;  
  
const int N = 110,inf = 0x3f3f3f3f;  
int n;  
int num[N];  
int dp[N][N];  
string s;  
  
inline void init(){  
    memset(dp,inf,sizeof(dp));  
    for (re int i = 1;i <= 9;i++) num[i] = 1;  
    for (re int i = 10;i <= 99;i++) num[i] = 2;  
    num[100] = 3;  
    for (re int i = 1;i <= n;i++) dp[i][i] = 1;  
}  
  
inline bool check(int l,int r,int x){  
    for (re int i = l + x;i <= r;i++){  
        int id = l + (i - l) % x;  
        if (s[i] != s[id]) return false;  
    }  
    return true;  
}  
  
int main(){  
    ios::sync_with_stdio(0);  
    cin.tie(0);  
    cout.tie(0);  
    cin >> s;  
    n = s.length();  
    s = ' ' + s;  
    init();  
    for (re int l = 2;l <= n;l++){  
        for (re int i = 1;i + l - 1 <= n;i++){  
            int j = i + l - 1;  
            for (re int k = i;k < j;k++) dp[i][j] = min(dp[i][j],dp[i][k] + dp[k + 1][j]);  
            for (re int k = 1;k < l;k++){  
                if (l % k == 0){  
                    int cnt = l / k;  
                    if (check(i,j,k)) dp[i][j] = min(dp[i][j],dp[i][i + k - 1] + num[cnt] + 2);  
                }  
            }  
        }  
    }  
    cout << dp[1][n];  
    return 0;  
}