计蒜客 青出于蓝胜于蓝

青出于蓝胜于蓝

题目描述

武当派一共有 nn 人，门派内 nn 人按照武功高低进行排名，武功最高的人排名第 11，次高的人排名第 22，… 武功最低的人排名第 nn。现在我们用武功的排名来给每个人标号，除了祖师爷，每个人都有一个师父，每个人可能有多个徒弟。

我们知道，武当派人才辈出，连祖师爷的武功都只能排行到 pp。也就是说徒弟的武功是可能超过师父的，所谓的青出于蓝胜于蓝。

请你帮忙计算每个人的所有子弟（包括徒弟的徒弟，徒弟的徒弟的徒弟….）中，有多少人的武功超过了他自己。

输入

输入第一行两个整数 n, p(1 \le n \le 100000, 1 \le p \le n)n,p(1≤n≤100000,1≤p≤n)。

接下来 n-1n−1 行，每行输入两个整数 u, v(1 \le u, v \le n)u,v(1≤u,v≤n)，表示 uu 和 vv 之间存在师徒关系。

输出

输出一行 nn 个整数，第 ii 个整数表示武功排行为 ii 的人的子弟有多少人超过了他。

行末不要输出多余的空格。

样例

样例输入

10 5
5 3
5 8
3 4
3 1
2 1
6 7
8 7
9 8
8 10
样例输出

0 0 2 0 4 0 1 2 0 0

题意

计算客这个蓝翔杯模拟赛有毒 题比正式赛难多了可惜都是模板题+原题，，，
黑心商业OJ

AC代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
#define LL long long
#define CLR(a,b) memset(a,(b),sizeof(a))

const int MAXN = 1e5+10;
vector<int> T[MAXN];
int vis[MAXN], res[MAXN];
int c[MAXN];
int n, root;

int lowbit(int x) {
    return x&(-x);
}
int query(int x) {
    int res = 0;
    while(x) {
        res += c[x];
        x -= lowbit(x);
    }
    return res;
}
void add(int x,int d) {
    while(x <= n) {
        c[x] += d;
        x += lowbit(x);
    }
}
void dfs(int u) {
    vis[u] = 1;
    res[u] = query(u);
    add(u,1);
    for(int i = 0; i < T[u].size(); i++) {
        int v = T[u][i];
        if(!vis[v])
            dfs(v);
    }
    res[u] = query(u)-res[u]-1;
}

int main() {
    ios::sync_with_stdio(false);
    while(cin >> n >> root) {
        CLR(vis,0);
        CLR(c,0);
        CLR(res,0);
        for(int i = 0; i < n-1; i++) {
            int u, v;
            cin >> u >> v;
            T[u].push_back(v);
            T[v].push_back(u);
        }
        dfs(root);
        for(int i = 1; i < n; i++) {
            cout << res[i] << ' ';
        }
        cout << res[n] << endl;
        for(int i = 1; i <= n; i++) T[i].clear();

    }
    return 0;
}