牛客多校碰到毒瘤括号dp题，题解记录一下。

K题目描述

题目分析

给定长度为n的括号序列（不保证合法性），求在此基础上生成的长度为m括号序列的方案数。

设 \(dp[i][j][k]\)表示插入括号的数量为 \(i\)、使用的原来的序列中的括号数量为 \(j\)，左括号比右括号多 \(k\) 时的方案数。那么最终答案为 \(dp[m-nl[n][0]\)。那么考虑如何设计状态转移：

首先枚举插入的括号数量，原来的括号序列和左括号比右括号多的数量。 如果目前枚举到的括号为左括号，并且使用的原括号的数量\(< n\)，就可以将该括号放入最终序列中，即为： \[ dp[i][j+1][k+1]=(dp[i][j+1][k+1]+dp[i][j][k])\%mod \] 如果此时枚举到的是一个右括号，并且\(k>0\)，即左括号的数量大于右括号的数量，并且使用的原括号的数量\(<n\)，就将该右括号放入最终序列： \[ dp[i][j+1][k-1]=(dp[i][j+1][k-1]+dp[i][j][k])\%mod \] 如果使用的括号数量为 \(n\)，或当前枚举到右括号，则可以插入左括号： \[ dp[i+1][j][k+1]=(dp[i+1][j][k+1]+dp[i][j][k])\%mod \] 当 \(k>0\) 时，如果使用原序列括号的数目为 \(n\)，或当前枚举到左括号，则可以插入右括号： \[ dp[i+1][j][k-1]=(dp[i+1][j][k-1]+dp[i][j][k])\%mod \]

标程

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#include <bits/stdc++.h>
#pragma gcc optimize(2)
//#define int long long
#define endl '\n'
using namespace std;

const int N = 2e5 + 10, mod = 1e9 + 7;
int dp[220][220][220];

inline void solve(){
    //memset(dp, 0, sizeof(dp));
    int n, m; cin >> m >> n;
    string s; cin >> s;
    //if(n & 1 || (m - n) & 1) { cout << 0 << endl; return; }
    dp[0][0][0]=1;
    for (int i = 0; i <= n - m; ++i)
    for (int j = 0; j <= m; ++j)
        for (int k = 0; k <= n; ++k){
            if (s[j] == '(' && j < m)
                dp[i][j + 1][k + 1] = (dp[i][j + 1][k + 1] + dp[i][j][k]) % mod;
            else
                dp[i + 1][j][k + 1] = (dp[i + 1][j][k + 1] + dp[i][j][k]) % mod;
            if (k > 0){
                if (s[j] == ')' && j < m)
                    dp[i][j + 1][k - 1] = (dp[i][j + 1][k - 1] + dp[i][j][k]) % mod;
                else
                    dp[i + 1][j][k - 1] = (dp[i + 1][j][k - 1] + dp[i][j][k]) % mod;
            }
        }
    cout << dp[n - m][m][0] << endl;
    for(int i = 0; i <= n + 2; i++){
        for(int j = 0; j <= m + 2; j++){
            for(int k = 0; k <= n + 2; k++) dp[i][j][k] = 0;
        }
    }
}

signed main(){
    ios_base::sync_with_stdio(false), cin.tie(0);
    int t = 1; cin >> t;
    while(t--) solve();
    return 0;
}