求树的子树数目

题目是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;
}