区间DP,按照一定规则给括号涂色,问有多少种涂法
首先可以想到d[i][j]保存区间i~j的涂法个数,但是规则中对颜色的要求光是这么开处理不了
就用0代表无色,1代表红,2代表蓝,d[i][j][0][1]就代表区间i~j且最左端为无色,最右端为红色的涂法个数
对于区间i~j,i和j为一对括号和不是一对的情况分开处理
是一对括号的话就递归先处理d[i+1][j-1],然后在用这个结果处理d[i][j]
不是一对就找到i的配对p,处理d[i][p]和d[p+1][j],然后用结果处理d[i][j]
如果i+1==j的话,那么i,j肯定是一对括号(按照上面的查找规则就可以保证)
详见代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stack>
using namespace std;
typedef long long ll;
const int N=700+10;
const ll MOD=1000000007;
char s[N];
int a[N];
ll d[N][N][3][3];
void dp(int l,int r)
{
if(l+1==r)
{
d[l][r][0][1]=1;
d[l][r][1][0]=1;
d[l][r][0][2]=1;
d[l][r][2][0]=1;
return;
}
if(a[l]==r)
{
dp(l+1,r-1);
for(int i=0;i<3;i++)
{
for(int j=0;j<3;j++)
{
if(j!=1)
d[l][r][0][1]=(d[l][r][0][1]+d[l+1][r-1][i][j])%MOD;
if(i!=1)
d[l][r][1][0]=(d[l][r][1][0]+d[l+1][r-1][i][j])%MOD;
if(j!=2)
d[l][r][0][2]=(d[l][r][0][2]+d[l+1][r-1][i][j])%MOD;
if(i!=2)
d[l][r][2][0]=(d[l][r][2][0]+d[l+1][r-1][i][j])%MOD;
}
}
}
else
{
int p=a[l];
dp(l,p);
dp(p+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 q=0;q<3;q++)
{
if(!((k==1&&q==1)||(k==2&&q==2)))
d[l][r][i][j]=(d[l][r][i][j]+(d[l][p][i][k]*d[p+1][r][q][j])%MOD)%MOD;
}
}
}
}
}
}
int main()
{
while(~scanf("%s",s))
{
int n=strlen(s);
stack<int> st;
for(int i=0;i<n;i++)
{
if(s[i]=='(') st.push(i);
else if(s[i]==')')
{
a[i]=st.top();
a[st.top()]=i;
st.pop();
}
}
memset(d,0,sizeof(d));
dp(0,n-1);
ll ans=0;
for(int i=0;i<3;i++)
for(int j=0;j<3;j++)
ans=(ans+d[0][n-1][i][j])%MOD;
printf("%lld\n",ans);
}
return 0;
}