CSP-S 2021 T2 括號序列 題解

題面

考場上想了 114514 年都沒想出怎么直接不算重，然后寫了個容斥減掉算重的調了 1919810 年還沒調出來（貌似這樣做不行？）/kk

看到 \(n\le 500\) 一眼區間 dp。

設 \(f_{i,j}\) 表示區間 \([i,j]\) 成為合法括號序列的答案。轉移的話直接按照題目的方式轉移就是了。

但是很明顯像 ()()() 這樣的序列會算重。。。

於是考慮對於 AB、ASB 這樣的轉移，強制讓它只從 \([i,\) 第一個左括號匹配的右括號的位置 \(]\) 轉移過來。

可以考慮設 \(g_{i,j}\) 表示區間 \([i,j]\) 只由 ()、(S)、(A)、(SA)、(AS) 轉移過來的合法序列個數。

這樣就解決會算重的問題了。

實現時務必注意細節。

#include <bits/stdc++.h>
#define DEBUG fprintf(stderr, "Passing [%s] line %d\n", __FUNCTION__, __LINE__)
#define File(x) freopen(x".in","r",stdin); freopen(x".out","w",stdout)
#define DC int T = gi <int> (); while (T--)
#define fi first
#define se second
#define pb push_back
#define mp make_pair

using namespace std;

typedef long long LL;
typedef pair <int, int> PII;
typedef pair <int, PII> PIII;

template <typename T>
inline T gi()
{
    T f = 1, x = 0; char c = getchar();
    while (c < '0' || c > '9') {if (c == '-') f = -1; c = getchar();}
    while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return f * x;
}

const int INF = 0x3f3f3f3f, N = 503, M = N << 1, mod = 1000000007;

int n, K;
int f[N][N], g[N][N], sum[N][N];
char s[N];
bool p[N][N];
int mn[N];

inline bool Can(int pos, char c) {return s[pos] == c || s[pos] == '?';}
inline void Add(int &x, int y) {x = (x + y) % mod;}

int main()
{
	//File("");
	n = gi <int> (), K = gi <int> ();
	scanf("%s", s + 1);
	memset(mn, 0x3f, sizeof mn);
	for (int i = 1; i <= n; i+=1)
		for (int j = i; j <= min(i + K - 1, n); j+=1)
			if (Can(j, '*')) p[i][j] = true, mn[j] = min(mn[j], i);
			else break;
	for (int len = 2; len <= n; len+=1)
		for (int i = 1; i + len - 1 <= n; i+=1)
		{
			int j = i + len - 1;
			sum[i][j] = sum[i][j - 1];
			if (!Can(i, '(') || !Can(j, ')')) continue;
			if (len == 2) {f[i][j] = g[i][j] = 1; Add(sum[i][j], f[i][j]); continue;}
			if (p[i + 1][j - 1]) f[i][j] = g[i][j] = 1;
			for (int k = i; k < j; k+=1)
				Add(g[i][j], 1ll * f[i][k] * g[k + 1][j] % mod);
			for (int l = i + 1; l < j; l+=1)
				if (mn[l] != 0x3f3f3f3f)
				{
					int tmp = max(mn[l], i + 1);
					Add(g[i][j], 
						1ll * g[l + 1][j] * 
						((sum[i][l - 1] + mod - sum[i][tmp - 2]) % mod) % mod);
				}
			Add(f[i][j], g[i + 1][j - 1]), Add(g[i][j], g[i + 1][j - 1]);
			for (int k = i + 1; k <= min(i + K, j - 2); k+=1)
				if (p[i + 1][k]) 
					Add(f[i][j], g[k + 1][j - 1]), Add(g[i][j], g[k + 1][j - 1]);
				else break;
			for (int k = max(i + 2, j - K); k < j; k+=1)
				if (p[k][j - 1])
					Add(f[i][j], g[i + 1][k - 1]), Add(g[i][j], g[i + 1][k - 1]);
			Add(sum[i][j], f[i][j]);
		}
	printf("%d\n", g[1][n]);
	return 0;
}