CodeForces 149 D.Coloring Brackets（区间DP）

Description
给出一个长度为n的合法括号序列，要求给其中一些括号染色，颜色只有两种——红和蓝，染色需要满足下面两个条件：
1.一个括号要么不染色，要么染成红色，要么染成蓝色
2.对一对匹配的左右括号，只能有一个染色
3.相邻的括号如果都染色则不能同色
问染色方案数
Input
一个长度为n的字符串表示一个合法的括号序列s（2<=|s|<=700）
Output
输出染色方案数，结果模1e9+7
Sample Input
(())
Sample Output
12
Solution
区间DP，dp[l][r][x][y]表示给区间[l,r]染好颜色且左端点颜色是x，右端点颜色时y的方案数，0,1,2分别表示不染色，红色和蓝色
如果l和r是一对匹配的括号，那么可以脱掉这层括号求出dp[l+1][r-1][x][y]，枚举l+1和r-1两个位置的括号颜色i和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;
如果l和r不匹配，则可以找到与l匹配的括号mid，求出dp[l][mid]和dp[mid+1][r]后，枚举l,mid,mid+1,r这四个位置的括号颜色分别为i,j,x,y，进而有转移方程
if(j*x==0||j!=x)dp[l][r][i][y]=(dp[l][r][i][y]+dp[l][mid][i][j]*dp[mid+1][r][x][y]%mod)%mod;
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 777
#define mod 1000000007
ll dp[maxn][maxn][3][3];
char s[maxn];
int belong[maxn];
void dfs(int l,int r)
{
    if(l+1==r)
    {
        dp[l][r][0][1]=dp[l][r][0][2]=dp[l][r][1][0]=dp[l][r][2][0]=1;
        return ;
    }
    if(belong[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=belong[l];
        dfs(l,mid),dfs(mid+1,r);
        for(int i=0;i<3;i++)
            for(int j=0;j<3;j++)
                for(int x=0;x<3;x++)
                    for(int y=0;y<3;y++)
                    {
                        if(j*x!=0&&j==x)continue;
                        dp[l][r][i][y]=(dp[l][r][i][y]+dp[l][mid][i][j]*dp[mid+1][r][x][y]%mod)%mod;
                    }
    }
}
int main()
{
    while(~scanf("%s",s))
    {
        int n=strlen(s);
        for(int i=0;i<n;i++)
            if(s[i]=='(')
            {
                int num=0;
                for(int j=i;j<n;j++)
                {
                    if(s[j]=='(')num++;
                    else num--;
                    if(num==0)
                    {
                        belong[i]=j;
                        break;
                    }
                }
            }
        memset(dp,0,sizeof(dp));
        dfs(0,n-1);
        ll ans=0;
        for(int i=0;i<3;i++)
            for(int j=0;j<3;j++)
                ans=(ans+dp[0][n-1][i][j])%mod;
        printf("%I64d\n",ans);
    }
    return 0;
}