计蒜客蓝桥杯模拟赛---青出于蓝而胜于蓝

题目链接

这道题综合运用了搜索和树状数组

原理如下

先DFS一下，搜索一整棵树，将树的每一个节点标记上时序（即DFS搜索时候的顺序）

记录两个，一个是往下搜的时序，一个是往回回溯的时序，这样这两个标记可以表示这个节点的全部子节点有多少个，有哪些

节点（比如标记为  1 3 ，那么子节点就有2 3）

建立一个空的树状数组。

然后从武当派的第一名开始，将人一个一个的添加到树状数组中

每次添加一个人，那么就把这个人的师傅的树状数组的C[i]加上1（包括师傅的师傅，师傅的师傅的师傅...直到祖师爷）

查询这个人有多少徒弟超过了自己，那么就用回溯的标记减去下搜的标记

#include<bits/stdc++.h>
using namespace std;
const int mmax = 100010;
struct node//存图
{
    int en,next;
} E[2 * mmax];
int p[mmax];
int num;
void init()
{
    memset(p,-1,sizeof p);
    num = 0;
}
void add(int st,int en)
{
    E[num].en = en;
    E[num].next = p[st];
    p[st] = num++;//
}
int l[mmax],r[mmax];
int times;
void dfs(int u,int fa)
{
    l[u] = ++times;//遍历时计算遍历次序
    for(int i = p[u]; i+1; i = E[i].next)
    {
        int v = E[i].en;
        if(v != fa)
        {
            dfs(v,u);
        }
    }
    r[u] = times;
}
int n,f;
int C[mmax];
void up(int x,int v)//树状数组的添加，往l[x]的后面添加1
{
    for(int i = x; i <= n; i += (i & -i))
    {
        C[i] += v;
    }
}
int sum(int x)//树状数组计算前和
{
    int res = 0;
    for(int i = x; i > 0; i -= (i & -i))
    {
        res += C[i];
    }
    return res;
}

int main()
{
    init();
    scanf("%d%d",&n,&f);
    for(int i = 0; i < n - 1; ++i)
    {
        int u,v;
        cin >> u >> v;//输入
        add(u,v);//双向存图
        add(v,u);
    }
    times = 0;//times表示存图时的时序
    dfs(f,-1);
    for(int i = 1; i <= n; ++i)
    {
        cout << sum(r[i]) - sum(l[i]) << " \n"[i == n];
        up(l[i],1);
    }
    return 0;
}