题目描述
给你一个字符串,并且告诉你这个是一棵树的一次 DFS 遍历序列,使用颜色标记节点,并且不保证每个节点颜色不同,也就是存在相同颜色的节点。
你需要输出合法的树结构存在几个?
Solition
首先这是一个类 DFS 序,那么是不是可以得到结论,使用区间 [ l, r] 构造的树,一定会存在一些分割点 k ,使得 s[l] = s[k],只有这样才能满足DFS的回溯条件。
那么我们是不是就可以使用区间DP去处理这个问题了。如果假设 dp[i][j] 代表以i位置为根,j位置结束的合法树结构数目。那么初值就是dp[i][i]=1。
接下来只需要枚举奇数区间,因为不存在那个合法的DFS回溯会存在偶数长度。
如果区间端点都不相同,直接continue掉这个区间,直到找到一个区间是符合要求的。那么我们就可以去枚举分割点了。
但是这个分割点如果只有s[r]的时候。也就是ABA的情况,只有一颗子树。那么我初步直接让 dp[l][r] = dp[l+1][r-1]就可以处理掉这个情况。
接着在 [ l+2, r-2]区间去枚举可能正确的分割点累加(左边一个子树乘上右边一个子树的情况)即可,左子数的根是 l+1, 右子树的根是分割点 k
例如 ABABA 下标1开始的化,可以枚举3作为分割点,dp[1][5]+= dp[2][2] * dp[3][5],相当于先序遍历,依次变换最左边的子树的形状,保证答案不重复。
下面也有我之前写的记忆化搜索,区间dp和记忆化搜索都是类似的。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
inline void print(ll x, int op = 10) { if (!x) { putchar('0'); if (op) putchar(op); return; } char F[40]; ll tmp = x > 0 ? x : -x; if (x < 0)putchar('-'); int cnt = 0; while (tmp > 0) { F[cnt++] = tmp % 10 + '0'; tmp /= 10; } while (cnt > 0)putchar(F[--cnt]); if (op) putchar(op); }
const int MOD = 1e9;
const int N = 300 + 7;
ll dp[N][N];
char s[N];
int main() {
int T = 1;
//T = read();
while (T--) {
scanf("%s", s + 1);
int n = strlen(s + 1);
for (int i = 1; i <= n; ++i) dp[i][i] = 1;
for (int len = 3; len <= n; len += 2) {
for (int l = 1, r = l + len - 1; r <= n; ++l, ++r) {
if (s[l] != s[r]) continue;
dp[l][r] = dp[l + 1][r - 1];
for (int k = l + 2; k <= r - 2; ++k)
if (s[l] == s[k])
dp[l][r] = (dp[l][r] + dp[l + 1][k - 1] * dp[k][r]) % MOD;
}
}
print(dp[1][n]);
}
return 0;
}
/*
#include <bits/stdc++.h>
using namespace std;
#define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
typedef long long ll;
const int N = 300 + 7;
const int MOD = 1e9;
ll dp[N][N];
char s[N];
ll dfs(int l, int r) {
if (l > r) return 0;
if (l == r) return 1;
if (dp[l][r] != -1) return dp[l][r];
dp[l][r] = 0;
for (int i = l + 1; i < r; ++i)
if (s[l] == s[i + 1])
(dp[l][r] += dfs(l + 1, i) * dfs(i + 1, r) % MOD) %= MOD;
return dp[l][r];
}
int main() {
memset(dp, -1, sizeof(dp));
scanf("%s", s + 1);
printf("%lld\n", dfs(1, strlen(s + 1)));
return 0;
}
*/ 
京公网安备 11010502036488号