CodeForces - 149D Coloring Brackets(区间dp)

    xiaoxiao2021-03-25  99

    题目链接

    这是一道区间dp的题目……怎么说呢,让我认识到我之前对区间dp的理解太浅显和模式化了,总觉得就是枚举每个区间段就可以了, ==确实是我刷的题还太少。 这道题目也是对于这些颜色该怎么处理无从下手,看完题解才觉得好巧妙。dp[rl[r][prel][prer],用这个数组来表示每个区间的状态,prel prer即代表这个区间外的紧挨的两个括号的颜色,有三种状态0 1 2,0为不染色,1为红色,2为蓝色。

    而对于每一个区间,要怎么划分它的子问题呢,并不是可以看成任意两段的组合,这两段的问题一定是不重叠的,所以,只有与左括号或者右括号的匹配的那个地方才是可以划分的,由于这两种情况下一段一段的分下去最后结果一定是一样的,所以取哪一个都可以。 然后通过枚举括号的颜色,来继续求解问题。 最后的结果即为dp[1][n][0][0],相当于给外围加了两个没有颜色的括号,并不影响结果。

    #include<cstdio> #include<cstdlib> #include<cstring> #include <cmath> #include<iostream> #include<algorithm> #include <vector> #include <bitset> #include <stack> #define maxn 710 #define ll long long #define MEM(x,num) memset(x,num,sizeof(x)) #define inf 0x3f3f3f3f #define mod 1000000007 using namespace std; char s[maxn]; ll dp[maxn][maxn][3][3]; //0不染色 1红色 2蓝色 int match[maxn]; int n; void init() { stack<int> st; for(int i=1;i<=n;i++) { if(s[i]=='(') st.push(i); else { int a=st.top(); st.pop(); match[a]=i; match[i]=a; } } } ll solve(int l,int r,int prel,int prer) //prel prer分别代表l r之前的那个区间的颜色 { ll & ans=dp[l][r][prel][prer]; if(ans!=-1) return ans; if(l>=r) return ans=1; ans=0; int x=match[r]; for(int le=0;le<3;le++) //枚举x的颜色 for(int ri=0;ri<3;ri++) if(ri&&!le||le&&!ri) { if(ri&&ri==prer) continue; if(x==l&&le&&le==prel) continue; ans=(ans+solve(l,x-1,prel,le)*solve(x+1,r-1,le,ri))%mod; } return ans; } int main() { scanf("%s",s+1); memset(dp,-1,sizeof(dp)); n=strlen(s+1); init(); cout<<solve(1,n,0,0)<<endl; return 0; }
    转载请注明原文地址: https://ju.6miu.com/read-23782.html

    最新回复(0)