CodeForces-149D Coloring Brackets(区间dp)

D - Coloring Brackets

  CodeForces - 149D 

题意：给一个合法的括号串，然后问这串括号有多少种涂色方案，涂色要求为：

①每个括号只有三种选择：涂红色，涂蓝色，不涂色。
②每对括号有且仅有其中一个被涂色。
③相邻的括号不能涂相同的颜色，但是相邻的括号可以同时不涂色。

写了这道题我才知道区间dp还能用递归来写的，感觉之前想用循环的写法好蠢
题解：区间dp
对于区间[l,r]一共只有3种情况：
①l+1==r，因为括号串是合法的，这种情况一定是匹配成功的
②l与r匹配，再递归搜索[l+1,r-1]
③l与r不匹配，找到与l匹配的mid，mid一定在区间[l,r]内，再搜索[l,mid],[mid+1,r]

这样很容易想到，开设的数组dp[l][r][i][j]表示区间[l,r]左右端分别染色为i和j,其中0表示未染色，1表示红色，2表示蓝色

#include<string.h>
#include<cstdio>
#include<algorithm>
#include<stack>
using namespace std;
typedef long long LL;
const LL mod = 1000000007;
const int MX = 705;
LL dp[MX][MX][3][3];
int match[MX];
char a[MX];
void get_match(){
    stack<int>s;
    memset(match,-1,sizeof(match));
    int len=strlen(a);
    for(int i=0;i<len;i++){
        if(a[i]=='(') s.push(i);
        else{
            int l=s.top();s.pop();
            match[l]=i;
            match[i]=l;
        }
    }
}
//递归解法
void dfs(int l,int r){
    if(l+1==r){
        dp[l][r][0][1]=1;
        dp[l][r][0][2]=1;
        dp[l][r][1][0]=1;
        dp[l][r][2][0]=1;
    }
    else if(match[l]==r){
        dfs(l+1,r-1);
        for(int i=0;i<3;i++)
            for(int j=0;j<3;j++){
                if(i!=1) dp[l][r][1][0]=(dp[l][r][1][0]+dp[l+1][r-1][i][j])%mod;
                if(i!=2) dp[l][r][2][0]=(dp[l][r][2][0]+dp[l+1][r-1][i][j])%mod;
                if(j!=1) dp[l][r][0][1]=(dp[l][r][0][1]+dp[l+1][r-1][i][j])%mod;
                if(j!=2) dp[l][r][0][2]=(dp[l][r][0][2]+dp[l+1][r-1][i][j])%mod;
            }
    }
    else{
        int mid=match[l];
        dfs(l,mid);
        dfs(mid+1,r);
        for(int i=0;i<3;i++)
            for(int j=0;j<3;j++)
                for(int k=0;k<3;k++)
                    for(int t=0;t<3;t++){
                        if((k==1&&t==1)||(k==2&&t==2)) continue;
                        dp[l][r][i][j]=(dp[l][r][i][j]+dp[l][mid][i][k]*dp[mid+1][r][t][j])%mod;
                    }
    }
}
//循环解法
void solve(int n){
    for(int i=0;i<n;i++){
        dp[i][i+1][0][1]=dp[i][i+1][0][2]=dp[i][i+1][1][0]=dp[i][i+1][2][0]=1;
        if(match[i]!=i+1) dp[i][i+1][0][0]=dp[i][i+1][1][2]=dp[i][i+1][2][1]=1;
    }
    for(int l=1;l<n;l++){
        for(int i=0;i+l<n;i++){
            int j=i+l;
            if(j==match[i]){
                for(int x=0;x<3;x++)
                    for(int y=0;y<3;y++){
                        if(x!=1) dp[i][j][1][0]=(dp[i][j][1][0]+dp[i+1][j-1][x][y])%mod;
                        if(x!=2) dp[i][j][2][0]=(dp[i][j][2][0]+dp[i+1][j-1][x][y])%mod;
                        if(y!=1) dp[i][j][0][1]=(dp[i][j][0][1]+dp[i+1][j-1][x][y])%mod;
                        if(y!=2) dp[i][j][0][2]=(dp[i][j][0][2]+dp[i+1][j-1][x][y])%mod;
                    }
            }
            else{
                int mid=match[i];
                for(int k=0;k<3;k++)
                    for(int t=0;t<3;t++)
                        for(int x=0;x<3;x++)
                            for(int y=0;y<3;y++){
                                if((x==1&&y==1)||(x==2&&y==2)) continue;
                                dp[i][j][k][t]=(dp[i][j][k][t]+dp[i][mid][k][x]*dp[mid+1][j][y][t])%mod;
                            }
            }
        }
    }
}
int main() {
   // freopen("in.txt","r",stdin);
    while(~scanf("%s",a)){
        memset(dp,0,sizeof(dp));
        get_match();
        int len=strlen(a);
        dfs(0,len-1);
       // solve(len);
        LL ans=0;
        for(int i=0;i<3;i++)
            for(int j=0;j<3;j++)
                ans=(ans+dp[0][len-1][i][j])%mod;
        printf("%I64d\n",ans);
    }
    return 0;
}