CodeForces 149D Coloring Brackets(区间DP)

最新推荐文章于 2021-05-22 07:32:54 发布

原创最新推荐文章于 2021-05-22 07:32:54 发布 · 504 阅读

1 ·

CC 4.0 BY-SA版权

高效算法-动态规划专栏收录该内容

147 篇文章

订阅专栏

本文介绍了一种利用区间动态规划解决括号染色问题的方法。通过预处理匹配括号的位置信息，并使用四维DP数组记录每种状态下的方案数，最终实现了对给定括号串的有效染色。

先预处理出每个左括号对应匹配的右括号位置，然后进行区间DP，dp[l][r][lc][rc]表示区间L，R，左边禁止lc颜色，右边禁止rc颜色的情况数，然后进行区间DP即可

代码：

#include <cstdio>
#include <cstring>
#include <stack>
#include <algorithm>
using namespace std;

typedef long long ll;
const int N = 705;
const int MOD = 1000000007;

const int d[4][2] = {0, 1, 0, 2, 1, 0, 2, 0};

char s[N];
int to[N], dp[N][N][3][3];

int dfs(int l, int r, int lc, int rc) {
	int &ans = dp[l][r][lc][rc];
	if (ans != -1) return ans;
	ans = 0;
	if (l > r) return ans = 1;
	int mid = to[l];
	for (int i = 0; i < 4; i++) {
		if (lc && lc == d[i][0]) continue;
		if (rc && mid == r && d[i][1] == rc) continue;
		ans = (int)(((ll)ans + (ll)dfs(l + 1, mid - 1, d[i][0], d[i][1]) * dfs(mid + 1, r, d[i][1], rc) % MOD) % MOD);
	}
	return ans;
}

int main() {
	while (~scanf("%s", s + 1)) {
		int n = strlen(s + 1);
		stack<int> S;
		memset(dp, -1, sizeof(dp));
		for (int i = 1; i <= n; i++) {
			if (S.empty()) S.push(i);
			else {
				if (s[S.top()] == '(' && s[i] == ')') {
					to[S.top()] = i;
					S.pop();
				} else S.push(i);
			}
		}
		printf("%d\n", dfs(1, n, 0, 0));
	}
	return 0;
}