CF149D Coloring Brackets 区间DP+DFS

用两种颜色给括号进行染色,要保证每对匹配的括号中刚好只有一个括号被染色,并且相邻的括号染色不能相同。求染色方案数。

思路:区间DP,用 f [ i ] [ j ] [ a ] [ b ] f[i][j][a][b] f[i][j][a][b] 表示区间 [ i , j ] [i,j] [i,j] 中左右染色分别为 a , b a,b a,b 的情况下的染色方案数,其中 a , b ∈ { 0 , 1 , 2 } a,b\in\{0,1,2\} a,b{0,1,2} ,用 0 0 0 表示不染色,另外两个数字表示染不同的两种颜色。

首先得到括号序列中每个括号之间的匹配对象,然后用 DFS 进行记忆化搜索。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<stack>
#define M 1000000007
#define MAXN 800
using namespace std;
typedef long long ll;
char s[MAXN];
int n,m[MAXN];ll f[MAXN][MAXN][3][3];
stack<int> st;
void dfs(int l,int r){
    if(l+1==r)f[l][r][0][1]=f[l][r][0][2]=f[l][r][1][0]=f[l][r][2][0]=1;
    else if(m[l]==r){
        dfs(l+1,r-1);
        for(int i=0;i<=2;i++)
            for(int j=0;j<=2;j++){
                if(j!=1)f[l][r][0][1]=(f[l][r][0][1]+f[l+1][r-1][i][j])%M;
                if(j!=2)f[l][r][0][2]=(f[l][r][0][2]+f[l+1][r-1][i][j])%M;
                if(i!=1)f[l][r][1][0]=(f[l][r][1][0]+f[l+1][r-1][i][j])%M;
                if(i!=2)f[l][r][2][0]=(f[l][r][2][0]+f[l+1][r-1][i][j])%M;
            }
    }else{
        dfs(l,m[l]);dfs(m[l]+1,r);
        for(int i=0;i<=2;i++)
            for(int j=0;j<=2;j++)
                for(int p=0;p<=2;p++)
                    for(int q=0;q<=2;q++){
                        if((j==1&&p==1)||(j==2&&p==2))continue;
                        f[l][r][i][q]=(f[l][r][i][q]+f[l][m[l]][i][j]*f[m[l]+1][r][p][q]%M)%M;
                    }
    }
}
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    scanf(" %s ",s);
    n=strlen(s);
    for(int i=0;i<n;i++){
        if(s[i]=='(')st.push(i);
        else m[st.top()]=i,m[i]=st.top(),st.pop();
    }
    dfs(0,n-1);
    ll res=0;
    for(int i=0;i<=2;i++)
        for(int j=0;j<=2;j++)
            res=(res+f[0][n-1][i][j])%M;
    printf("%lld\n",res);
    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值