码迷,mamicode.com
首页 > 其他好文 > 详细

codeferces 149D Coloring Brackets 区间dp

时间:2015-04-30 08:55:37      阅读:127      评论:0      收藏:0      [点我收藏+]

标签:acm   codeforces   区间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;
}


codeferces 149D Coloring Brackets 区间dp

标签:acm   codeforces   区间dp   

原文地址:http://blog.csdn.net/xuelanghu407/article/details/45380069

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!