hust 1428【母函数】

http://acm.hust.edu.cn/thx/problem.php?id=1428

#include<iostream>
#include<cstdio>
#include<string.h>
#include<vector>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<algorithm>
using namespace std;
#define inf 0x7fffffff
vector<int> v[50005];
int a[50005];
int dfs(int x)
{
    int i,j;
    a[x]+=v[x].size();
    for(i=0;i<v[x].size();i++)
    {
        j=v[x][i];
        a[x]+=dfs(j);
    }
    return a[x];
}

int main()
{    
    int t;
    scanf("%d",&t);
    for(int ca=1;ca<=t;ca++)
    {
        int n,m;
        scanf("%d",&n);
        int i,j,k;
        for(i=0;i<=n;i++)
        v[i].clear(),a[i]=0;
        for(i=1;i<n;i++)
        {
            int num;
            scanf("%d",&num);
            v[num-1].push_back(i);
        }
        dfs(0);
        printf("Case #%d: ",ca);
        printf("%d",a[0]);
        for(i=1;i<n;i++)
        printf(" %d",a[i]);
        printf("/n");
    }
    return 0;
}