51Nod-1353-树

ACM模版

描述

题解

这个题做出来的人很少，看了看官方题解，如下：

我们令 $dp[i][j]$ 表示以 i 为根且当前联通块大小为 k 的方案总数，特别的，$dp[i][0]$ 表示割点当前点与其父亲是棵平衡树的方案总数。
对于 u 的一个孩子 v 可以得到转移方程 $dp[u][j + k] = dp[u][j] * dp[v][k]$
另外 $dp[u][0] = Σdp[u][j] (j >= 题目给定的 k )$
这样乍看是 $n^3$ 的，有一个技巧可以做到 $n^2$ 即每次 dp 时，只枚举当前 u 所在子树的大小，每当枚举到它的其中孩子时，当前 u 所在子树的大小加上它孩子为根的子树的大小。可以理解为每一个点对只被枚举到一次。
最后答案即为 $dp[root][0]$。

这个题虽然有官方题解，但是看着我也是一脸懵逼，只是清楚的知道这是一个树归，思前想后，于是找到了一个 AC 代码研究了一番，大致算是明白了，但是状态转移方程依然是似懂非懂……如果哪个大牛有高见，烦指教~~~

暂且说一下我的理解，因为这里是平衡树，所以没有确定几个分叉，只是确定高度差不大于一，所以我的理解是，一棵以 u 为根的树，无论去哪一棵子树，并且这个子树的根是 u 的孩子，那么最后一定都满足平衡，本着这样的思路，利用乘法原理可以得到如上转移方程。

不知我的理解是否有偏差，如果有的话，烦请告知！

代码

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;

const int MAXN = 2222;
const int MOD = 1e9 + 7;

int n, k;
int tot = 0;
int pre[MAXN];
int dp[MAXN][MAXN];
int s[MAXN];

struct edge
{
    int next;
    int to;
} e[MAXN << 1];

void AND_MOD(int &x, int y)
{
    x += y;
    if (x >= MOD)
    {
        x -= MOD;
    }
}

void dfs(int u, int fa)
{
    dp[u][s[u] = 1] = 1;
    for (int it = pre[u]; it; it = e[it].next)
    {
        if (e[it].to == fa)
        {
            continue;
        }
        dfs(e[it].to, u);
        for (int i = s[u]; i > 0; i--)
        {
            for (int j = s[e[it].to]; j > 0; j--)
            {
                AND_MOD(dp[u][i + j], (long long)dp[u][i] * dp[e[it].to][j] % MOD);
            }
            dp[u][i] = (long long)dp[u][i] * dp[e[it].to][0] % MOD;
        }
        s[u] += s[e[it].to];
    }

    for (int i = k; i <= s[u]; i++)
    {
        AND_MOD(dp[u][0], dp[u][i]);
    }
}

int main()
{
    cin >> n >> k;

    int u, v;
    for (int i = 1; i < n; ++i)
    {
        scanf("%d%d", &u, &v);
        e[++tot] = (edge){pre[u], v};
        pre[u] = tot;
        e[++tot] = (edge){pre[v], u};
        pre[v] = tot;
    }

    dfs(1, -1);

    printf("%d\n", dp[1][0]);

    return 0;
}