（CSP2019模拟）DTOJ 4624. 树

题意

给定一棵 $n$ 个结点的树，共有 $q$ 次询问。

第 $i$ 次询问首先包含了三个数 $k_i,m_i,r_i$ ，接着给定了树上互不相同的 $k_i$ 个关键点 $a_{i,1},a_{i,2},\dots ,a_{i,k}$ 。对于第
$i$次询问，你需要回答当这棵树以 $r_i$ 为根时，你有多少种方案将这 $k_i$ 个点分为至多 $m_i$ 组，使得同一组内的任意两个不同的结点都不存在祖先关系。

对于第 $i$ 次询问，假设你一共将 $k_i$ 个点分成了 $p$ 组，那么分组的方案需要满足：

• 给出的 $k_i$ 个点中每个点属于且仅属于 $p$ 组中的其中一组。
• $p$ 组中的任意一组都要至少包含一个结点。

对于第 $i$ 次询问，两个分组的方案不同，当且仅当它们分出的组数不同，或者存在两个点 $a_{i,x},a_{i,y}(x\ne y)$ ，它
们在其中一种分组方案中被分到了同一组，在另一种分组方案中没有被分到同一组。
数据范围：
对于 20% 的数据，$n\le 10,\sum k_i\le 20,m_i\le \min \left(4,k_i\right)$ ；

对于 60% 的数据，$n\le 10^4,\sum k_i\le 10^4,m_i\le \min \left(50,k_i\right)$ ；

对于另外 20% 的数据，保证第 $i$ 条边是 $(1,i+1)$ ，并且对于每个询问 $r_i=1$；

对于 100% 的数据，$n\le 10^5,q\le 10^5,\sum k_i\le 10^5,1\le u,v\le n$ 。

对于每个询问， $1\le k_i,r_i\le n,1\le m_i\le \min \left(300,k_i\right)$。

题解

考场：
只会直观的DP，记$f[i][j]$为$i$子树内分成$j$组的方案数，转移时枚举、容斥一下，但效率是$O(n\times m^2)$的，就算用虚树优化也过不了，于是只拿了60的暴力。
正解：
若直接在树上做，难以避免枚举子树分成几组，转移时效率不优秀。
故考虑能否每次只加入一个点。发现若按照一定顺序一个个加入点，记$f[i][j]$为前$i$个分为$j$组，则每次转移为$f[i][j]=f[i-1][j-1]+f[i][j-1]\times x$，十分优秀。考虑$x$怎么算，对于不能有祖先关系，只考虑后代不能放到祖先的集合，即每个点加入时，不能放到有它的祖先的集合，而此时它的祖先已全都加入，且一定在不同集合内，于是直接减去祖先个数即可。而对于加入点的顺序，考虑若在一颗树上，则按照深度顺序，那么对于一个根和一些关键点，即为它到根的路径上关键点的个数，整个过程用树状数组维护即可。