猿辅导面试题 括号染色 动态规划 组合数学

题目描述：

给定一个长度为$n$的数组$a$和一个正整数$k$，对于$a$数组的描述如下：

• 如果$a[i]==0$，那么$i$位置可以填入左括号或者右括号，而且此处的括号颜色可以从$1,2,3,...,k$中选择；
• 如果$a[i]\ne 0$，那么$i$位置是一个左括号，并且颜色为$a[i]$。

一个合法的括号序列是指的是，括号需要进行匹配，同时相互匹配的括号的颜色必须相同。
例如：()那么这个括号序列如果两个括号的颜色相同，那么该序列就是合法。
返回一共有多少种合法的括号序列。
由于答案可能会很大，你需要输出答案对$10001$取模。
数据范围：$2\le n\le 5\times 10^3,1\le k\le 10^3,0\le a[i]\le k$

题解：

由于相互匹配的括号的颜色一定要相同，我们将已经确定的左括号个数记为$cnt1$，那么就有$cnt1$个右括号的颜色一定要与这$cnt1$个左括号的颜色配对。
有了上述结论，那么我们考虑一个还未染色的合法括号序列能够染出多少种合法序列。由上述结论可以知道：序列中有$2\times cnt1$的位置的颜色是确定的，那么剩下的序列有$\frac{n-2cnt1}{2}$对括号，而每对括号有$k$种染色方法，那么由乘法原理就可以知道一共可以染出$k^{\frac{n-2cnt1}{2}}$。经过上述的推导，我们发现每个未染色的合法括号序列能染出的合法序列的个数时相同的，所以本题就转换成了求有多少个不染色的合法序列。求出来个数之后，只需要在乘$k^{\frac{n-2cnt1}{2}}$即可得到一共有多少种合法的括号序列。
下面来考虑如何求未染色的合法序列有多少种，我们需要先思考一下合法序列需要满足的条件：

• 在任意的位置右括号的数量一定不能超过左括号的数量；
• 并且在结尾时，左右括号的个数相同。

可以看到一个括号序列合不合法实际上是和左右括号的差值有关，这个差值一定要大于等于$0$，并且在最后的时候需要等于$0$。有了这些信息，我们就可以通过动态规划来求解，状态的定义也就需要围绕位置和左右括号的差值了。
状态定义如下：$dp[i][j]$表示考虑$[0,i]$上满足每个位置的左括号均不少于右括号并且到$i$位置时，左括号比右括号多$j$个的括号序列个数。
有了上述的状态，很容易可以写出状态转移方程：

• $dp[i][j]=dp[i-1][j-1]ifa[i]\ne 0$；
• $dp[i][j]=dp[i-1][j-1]+dp[i-1][j+1]ifa[i]==0$，分别对应当前位置填上左右括号。

$dp[n-1][0]$就表示不染色的情况下合法括号序列个数。

代码:

#include <bits/stdc++.h>

const int MAXN = 5e3 + 10;
const int MOD = 10001;

using namespace std;

int n, k, cnt1, ans;
int a[MAXN], dp[MAXN][MAXN];

int quickPower(int a, int b, int mod)
{
    int res = 1;
    while(b) {
        if (b & 1) { res = res * a % mod; }
        a = a * a % mod;
        b >>= 1;
    }
    return res % mod;
}

int main()
{
    ios::sync_with_stdio(false);
    cin >> n >> k;
    for (int i = 0; i < n; i++) { cin >> a[i]; }
    if (n % 2 == 1) {
        cout << "0" << endl;
        return 0;
    }
    for (int i = 0; i < n; i++) {
        if (a[i] != 0) { cnt1++; }
    }
    ans = quickPower(k, (n - 2 * cnt1) / 2, MOD);
    dp[0][1] = 1;
    for (int i = 1; i < n; i++) {
        for (int j = 0; j <= i+1;j ++) {
            if (a[i] == 0) {
                dp[i][j] = ((j - 1 >= 0 ? dp[i - 1][j - 1] : 0) + (j + 1 <= i ? dp[i - 1][j + 1] : 0)) % MOD;
            } else {
                dp[i][j] = (j - 1 >= 0 ? dp[i - 1][j - 1] : 0) % MOD;
            }
        }
    }
    ans *= dp[n - 1][0];
    cout << ans << endl;
    return 0;
}