codeferces 149D Coloring Brackets 区间dp

http://codeforces.com/problemset/problem/149/D

题目大致意思是给你一串字符串，只有小括号，并且已经符合括号匹配规则，现在要给这些括号涂色，给出一些涂色规则，求涂色的方案数。

1： 括号要么不被涂色，要么被涂成蓝色，要么被涂成红色。

2：两个相互匹配的括号有且仅有一个被涂色。

3：相邻两个括号不可以有相同颜色。

这里当然也是想到对区 [l, r] 间进行dp，但是这里对颜色有依赖关系，所以还需要记入 l 和 r 颜色的状态，一开始打算只用一维记录两个点的颜色，后来发现我果然还是too young

于是 dp[l][r][a][b] 表示区间 [l, r] 上其颜色方案分别是 a 和 b (a, b 取值0~2，分别表示不涂色，涂蓝色，涂红色)。

现在分情况讨论，

对于区间 [l, r] 如果 l r 相互匹配的话(这里是在整体的串中匹配)，其子问题是区间 [l+1, r-1] 在涂色为合法的 i, j 状态的和。即 dp[i][j][a][b] = sum (dp[l+1][r-1][i][j]); 其中，涂色方案[i, j]相对于方案[a,b]合法。

而如果l r 不相互匹配的话，则 k 为 l 括号的匹配括号，那么区间被拆分成 [l, k] 和 [k+1, r] ,于是我们枚举 k 和 k+1的涂色状态分别为 i, j 。答案便为方案[a, i] 对于方案[j, b]合法的和。即dp[l][r][a][b] = sum(dp[l][k][a][i] *dp[k+1][r][j][b]);

 

每个位置的对应得匹配括号位置可以用栈模拟得出。然后对区间进行记忆化搜索。对于l 和 r 匹配的区间，还需要对其涂色方案判断是否合法(因为不匹配的区间不好判断，并且最终还是分成几个匹配区间进行判断，所以就不判断了)。还有，题目要求对 10^9 + 7 取模。

#include <stack>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const long long MOD = 1000000007LL;

stack<int > S;
 
int v[707];

void doit(string a) {
    for (int i=0; i<a.size(); i++) {
        if (a[i] == '(') {
            S.push(i);
        } else {
            int tmp = S.top();
            S.pop();
            v[tmp] = i;
            v[i] = tmp;
        }
    }
}

long long dp[707][707][3][3];

long long DP(int l, int r, int a, int b) {
    if (v[l] == r && ((a == b) || (a != 0 && b != 0))) {
        return dp[l][r][a][b] = 0;
    } 

    if (dp[l][r][a][b] != -1) {
        return dp[l][r][a][b] % MOD;
    }
    if (r - l == 1) {
        return dp[l][r][a][b] = 1;
    }

    long long tmp = 0;
    
    if (v[l] != r) {

        int k = v[l];

        for (int i=0; i<3; i++) {
            for (int j=0; j<3; j++) {
                if ((i == 0 && j == 0) || i != j) {
                    tmp = (tmp + DP(l, k, a, i) * DP(k+1, r, j, b)) % MOD;
                }
            }
        }
    } else {

        for (int i=0; i<3; i++) {
            for (int j=0; j<3; j++) {
                if ((a == 0 && j != b) || (i != a && b == 0)) {
                    tmp = (tmp + DP(l+1, r-1, i, j)) % MOD;
                }
            }
        }
    }

    return dp[l][r][a][b] = tmp % MOD;
}

int main () {
    string a;
    cin >> a ;

    doit(a);

    memset(dp, -1, sizeof(dp));

    long long ans = 0;

    for (int i=0; i<3; i++) {
        for (int j=0; j<3; j++) {
            ans = (ans + DP(0, a.size()-1, i, j)) % MOD;
        }
    }

    cout << ans % MOD << endl;

    return 0;
}