求树的子树数目

最新推荐文章于 2022-04-05 15:16:51 发布

发奋屠强

最新推荐文章于 2022-04-05 15:16:51 发布

阅读量2.3k

点赞数

分类专栏： DP&&记忆化搜索

本文链接：https://blog.csdn.net/hhaile/article/details/8589198

版权

该博客主要讨论了如何计算给定一棵树的所有子树数量，包括空树。提出使用f[i]表示以节点i为根的子树数目，通过递归公式f[a] = (f[a1]+1)*(f[a2]+1)*...*(f[ak]+1)进行记忆化搜索求解，并在最后加上1以考虑空树情况。

摘要由CSDN通过智能技术生成

题目是srm 570 div2的1000

题意转化一下就是给你一棵树，叫你求这颗树的子树有多少个，包括空树。

我们用f[i]来表示以节点i为根的树的数目。

首先如果只是一个节点，那么以它为根的树的数目为1.

如果以a为根的一棵树有k个儿子，那么

f[a] = (f[a1]+1)*(f[a2]+1)*(f[a3]+1)....

由此可用记忆化搜索将f[1]到f[n]全都求出来，再相加，最后再加上1,即为空时，为所求。

class CentaurCompanyDiv2 {
public:
	long long count(vector <int>, vector <int>);
};

long long f[55];
bool vis[55];
vector <int> e[55];
void dfs(int rt, int fa)
{
    int len = e[rt].size();
    if(len==0){vis[rt]=true; f[rt]=1;return;}
    for(int i = 0; i < len; i++)