求树的子树数目

来源：互联网发布：淘宝阶层编辑：程序博客网时间：2024/04/28 15:58

题目是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++)    {        int son = e[rt][i];        if(son!=fa)        {            if(!vis[rt])dfs(son, rt);            f[rt]*=f[son]+1;        }    }    vis[rt]=true;}long long CentaurCompanyDiv2::count(vector <int> a, vector <int> b) {    int len = a.size();    for(int i = 0; i < 55; i++)    {        e[i].clear();        f[i]=1;vis[i]=false;    }    for(int i = 0; i < len; i++)    {        e[a[i]].push_back(b[i]);        e[b[i]].push_back(a[i]);    }long long ans = 0;dfs(1,0);for(int i = 1; i <= len+1; i++)        {ans += f[i];}    return ans+1;}