cogs 2039. 树的统计

2039. 树的统计

★★   输入文件：counttree.in   输出文件：counttree.out   简单对比
时间限制：1 s   内存限制：128 MB

【题目描述】

关于树的统计问题有多种多样的版本，这里你需要解决一个比较简单的问题：对于一棵包含N个节点的有根树，将所有点从1到N编号后，对于每一个节点v，统计出以v为根的子树中有多少个点的编号比v小。

【输入格式】

输入第一行包含一个整数N，以下N行每行包含一个整数，其中第i行的整数表示编号为i的节点的父亲节点的编号，根的父亲节点编号为0。

【输出格式】

输出包含N行，其中第i行给出编号为i的节点的统计结果。

【样例输入】

3
2
3
0

【样例输出】

0 1 2

【提示】

在此键入。

【来源】

20%的数据1<=n<=1000

100%的数据1<=n<=100000

这题可以用循环水过的

我用的dfs序

#include<cstdio>
#include<vector>
using namespace std;

const int N=;
vector<int>vec[N];
int n,a;
int read(){
    int x=,f=;
    char c=getchar();
    while(c<''||c>''){if(c=='-')f=-;c=getchar();}
    while(c>=''&&c<=''){x=x*+c-'',c=getchar();}
    return x*f;}

int num=;
int head[N];
int tail[N];
int dfsx[N];

void dfs(int root)
{
    dfsx[++num]=root;head[root]=num;
    for(int i=;i<vec[root].size();i++)
    {
        dfs(vec[root][i]);
    }
    tail[root]=num;
}

int main()
{
    freopen("counttree.in","r",stdin);
    freopen("counttree.out","w",stdout);
    n=read();
    int r;
    for(int i=;i<=n;i++)
    {
        a=read();
        if(a==)
        r=i;
        vec[a].push_back(i);
    }
    dfs(r);
    int ans=;
    for(int i=;i<=n;i++)
    {
        ans=;
        for(int j=head[i]+;j<=tail[i];j++)
        {
            if(i>dfsx[j])ans++;
        }
        printf("%d\n",ans);
    }
    return ;
}