TopCoder SRM 570 Div2 1000 CentaurCompanyDiv2

这次的1000算简单的，就是统计有几个子树。
树形DP一波，$f[i]$表示以$i$为根的子树数量，枚举儿子，转移有三种种情况：

• 只连之前的儿子（$f[i]$）
  • 只连当前儿子（$f[v[i]]+1$）
  • 都连（$f[i]*(f[v[i]]+1)$）

  • 分别计数即可。

    #include <bits/stdc++.h>
    #define ll long long
    using namespace std;
    const int N=55;
    const int M=N<<1;
    int n,cnt;
    int head[N],Next[M],v[M];
    ll f[N];
    
    class CentaurCompanyDiv2 {
    public:
        long long count( vector <int> a, vector <int> b );
    };
    
    void add(int x,int y){
        Next[++cnt]=head[x];
        head[x]=cnt;
        v[cnt]=y;
    }
    
    void dfs(int x,int fa){
        for(int i=head[x];i!=-1;i=Next[i])
         if (v[i]!=fa){  
            dfs(v[i],x);
            f[x]*=(f[v[i]]+2);
            f[x]+=f[v[i]]+1;
         }
    }
    
    long long CentaurCompanyDiv2::count(vector <int> a, vector <int> b) {
        n=a.size()+1;cnt=0;
        memset(head,-1,sizeof head);
        for(int i=0;i<n-1;i++)
         add(a[i],b[i]),add(b[i],a[i]);
        memset(f,0,sizeof f);
        dfs(1,0);
        ll ans=0;
        for(int i=1;i<=n;i++) ans+=f[i];
        return ans+n+1;
    }