cogs2039树的统计 x

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

思路:

　　就是用搜索w

T代码:

#include <algorithm>
#include <iostream>
#include <cstring>
#include <queue>
#include <string>
#include <cstdio>

using namespace std;

inline int reads()
{
    int x=,f=;
    char ch=getchar();

    while(ch>'' || ch<'')
    {
        if(ch=='-') f=-;
        ch=getchar();
    }

    while(ch>='' && ch<='')
    {
        x=x*+ch-'';
        ch=getchar();
    }

    return x*f;
}

const int Ms = 1e5 + ;
int n,root;
int dad[Ms];
bool Vss[Ms];
int sum[Ms];
int deep[Ms];
int h[Ms],num;
queue<int>q;

struct AC{
    int to,next;
}t[Ms];

void ADD(int x,int y)
{
    t[++num].to=y;
    t[num].next=h[x];
    h[x]=num;
}

int ans;

void dfs(int u)
{                                        ///dfs the deep
    Vss[u]=true;
    int maxx=;
    for(int j=h[u];j;j=t[j].next)
    {
        int v=t[j].to;
        if(Vss[v]) continue;
        deep[v]=deep[u]+;                ///update
        dad[v]=u;                        ///u is v's dad
        dfs(v);                         ///continue dfs
        if(sum[v] > sum[maxx]) maxx=v;     ///update
        sum[u]+=sum[v]+;
    }
}

void QwQ(int now)
{
    ans=;
    int nowdeep=deep[now];
    for(int i=;i<now;i++)
    {
//        if(i==now) continue;
        if(ans>=sum[now]) break;
        if(deep[i]>nowdeep && ans<=sum[now])
        {
            if(dad[i]==now) ans++;
            else
            {
                int j=dad[i];
                while(deep[j] > deep[now])
                {
                    j=dad[j];
                    if(j==now) ans++;
                }
            }
        }
    }
}

int main()
{
    freopen("counttree.in","r",stdin);
    freopen("counttree.out","w",stdout);
    n=reads();
    int i=;                            ///fu chu zhi !!!
    int QAQ;
    int sss=n;
    while(sss--)
    {
        ///i's dad is dad[i]
        QAQ=reads();
        ADD(QAQ,i);
        ///find the root
        if(QAQ==) root=i;
        i++;
    }
    deep[root]=;
    sum[]=-;
    dfs(root);
    for(int i=;i<=n;i++)
    {
        QwQ(i);
        printf("%d\n",ans);
    }
    return ;
}

QAQ

A代码:

#include <algorithm>
#include <iostream>
#include <cstring>
#include <queue>
#include <string>
#include <cstdio>

using namespace std;

inline int reads()
{
    int x=,f=;
    char ch=getchar();

    while(ch>'' || ch<'')
    {
        if(ch=='-') f=-;
        ch=getchar();
    }

    while(ch>='' && ch<='')
    {
        x=x*+ch-'';
        ch=getchar();
    }

    return x*f;
}

const int Ms = 1e5 + ;
int n,root;
int now;
int h[Ms],num;

struct AC{
    int to,next;
}t[Ms];

void ADD(int x,int y)
{
    t[++num].to=y;
    t[num].next=h[x];
    h[x]=num;
}

int ans;

void dfs(int u)
{
    for(int j=h[u];j;j=t[j].next)
    {
        if(j<now) ++ans;
        dfs(j);                         ///continue dfs
    }
}

int main()
{
    freopen("counttree.in","r",stdin);
    freopen("counttree.out","w",stdout);
    n=reads();
    int QAQ,sss=n,i=;
    while(sss--)
    {
        ///i's dad is dad[i]
        QAQ=reads();
        ADD(QAQ,i);
        i++;
    }
    for(int i=;i<=n;++i)
    {
        now=i;
        ans=;
        dfs(i);
        printf("%d\n",ans);
    }
    return ;
}

23333