题目描述

给你一个字符串,并且告诉你这个是一棵树的一次 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;
}
*/