Codeforce 149D(区间dp+括号匹配)

链接：点击打开链接

题意：给一个给定括号序列，给该括号序列上色，上色有三个要求1.每个括号只有三种情况，不上色，上红色，上蓝色，2.每对括号必须只能给其中的一个上色3.相邻的两个不能上同色，可以都不上色，求所有的上色种数

代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const long long MOD=1000000007;
char s[705];
long long dp[705][705][3][3],a[705],b[705];
int main(){
long long i,j,k,l,p,x,y,xx,yy,ans,len;
while(scanf("%s",s+1)!=EOF){
p=1;
len=strlen(s+1);
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
memset(dp,0,sizeof(dp));
for(i=1;i<=len;i++){
if(s[i]=='(')
a[p++]=i;
else{
b[a[p-1]]=i;
b[i]=a[p-1];
p--;
}
}                                       //模拟栈进行括号匹配
for(i=1;i<len;i++)
if(b[i]==i+1){
dp[i][i+1][0][1]=dp[i][i+1][1][0]=1;
dp[i][i+1][0][2]=dp[i][i+1][2][0]=1;
}
for(l=2;l<=len;l++){
for(i=1;i<=len-l+1;i++){
j=i+l-1;
if(b[i]==j){
dp[i][j][0][1]+=(dp[i+1][j-1][0][0]+dp[i+1][j-1][1][0]+dp[i+1][j-1][2][0]+dp[i+1][j-1][0][2]+dp[i+1][j-1][1][2]+dp[i+1][j-1][2][2])%MOD;
dp[i][j][0][2]+=(dp[i+1][j-1][0][0]+dp[i+1][j-1][0][1]+dp[i+1][j-1][1][0]+dp[i+1][j-1][1][1]+dp[i+1][j-1][2][0]+dp[i+1][j-1][2][1])%MOD;
dp[i][j][1][0]+=(dp[i+1][j-1][0][0]+dp[i+1][j-1][0][1]+dp[i+1][j-1][2][0]+dp[i+1][j-1][0][2]+dp[i+1][j-1][2][1]+dp[i+1][j-1][2][2])%MOD;
dp[i][j][2][0]+=(dp[i+1][j-1][0][0]+dp[i+1][j-1][0][1]+dp[i+1][j-1][1][0]+dp[i+1][j-1][1][1]+dp[i+1][j-1][0][2]+dp[i+1][j-1][1][2])%MOD;
}                               //匹配的括号只能有一个染色并且颜色不能相同
else if(b[i]<j){
for(x=0;x<=2;x++)
for(y=0;y<=2;y++)
for(xx=0;xx<=2;xx++)
for(yy=0;yy<=2;yy++)
if(xx==0||yy==0||xx!=yy){
dp[i][j][x][y]+=(dp[i][b[i]][x][xx]*dp[b[i]+1][j][yy][y])%MOD;
dp[i][j][x][y]%=MOD;
}                           //处理相邻两个括号不能是相同颜色的情况
}
}
}
ans=0;
for(i=0;i<=2;i++)
for(j=0;j<=2;j++)
ans=(ans+dp[1][len][i][j])%MOD;
printf("%I64d\n",ans);
}
return 0;
}