计蒜客2018蓝桥A组模拟赛(一)青出于蓝胜于蓝(dfs序+线段树)

题目:

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

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

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

1 <= n <= 100000

分析:

这道题就是:给一棵带权树,求每个节点的子树上有多少个权值比它小的点。

显然,对每个节点遍历子树肯定会超时。

我们对这棵树dfs一次,按照dfs的顺序给每个点编号。这样,按照遍历的顺序我们就得到了一个线性序列。更重要的是,点 i i 的子树在线性序列中就紧跟点 i i 后面。所以我们如果知道它子树这一段的起点 l[i] l [ i ] 和终点 r[i] r [ i ] ,就可以用线段树解决。很显然,dfs的顺序得到的编号就是 l[i] l [ i ] 。在dfs的时候,某个点的子树遍历完了的时候,再记录一下 r[i] r [ i ]

这样,在dfs得到的线性序列中,点 i i 的子树就是 l[i] l [ i ] r[i] r [ i ]

我们从小排名到大排名更新和查询,对第 i i 名的人,比他名次小的人数 l[i] l [ i ] r[i] r [ i ] 的人数和。因为我们是按照从小到达查询更新,所以比 i i 排名大的点还没更新进树里。

代码:

#include 
using namespace std;
#define ms(a,b) memset(a,b,sizeof(a))
#define lson rt*2,l,(l+r)/2
#define rson rt*2+1,(l+r)/2+1,r
typedef unsigned long long ull;
typedef long long ll;
const int MAXN = 100005;
const double EPS = 1e-8;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
int tree[MAXN << 2], n, p, l[MAXN], r[MAXN], times;
vector<int>G[MAXN];

void init() {
    times = 0;
    ms(tree, 0);
}

void pushup(int rt) {
    tree[rt] = tree[rt << 1] + tree[rt << 1 | 1];
}

void dfs(int u, int fa) {
    l[u] = ++times;
    for (int i = 0; i < (int)G[u].size(); i++) {
        int v = G[u][i];
        if (v != fa) {
            dfs(v, u);
        }
    }
    r[u] = times;
}

int query(int L, int R, int rt, int l, int r) {
    if (L <= l && R >= r) {
        return tree[rt];
    }
    int ret = 0;
    if (L <= (l + r) / 2)   ret += query(L, R, lson);
    if (R > (l + r) / 2)    ret +=query(L,R,rson);
    return ret;
}

void update(int k, int rt, int l, int r) {
    if (l == r) {
        tree[rt] = 1;
        return;
    }
    if (k <= (l + r) / 2)   update(k, lson);
    else    update(k, rson);
    pushup(rt);
}

int main() {
    ios::sync_with_stdio(false);
    cin >> n >> p;
    init();
    for (int i = 0; i < n - 1; i++) {
        int u, v;
        cin >> u >> v;
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(p, -1);
    // for(int i=1;i<=n;i++){
    //  cout <
    // }
    for (int i = 1; i <= n; i++) {
        cout << query(l[i], r[i], 1, 1, n) << " \n"[i == n];
        update(l[i], 1, 1, n);
    }

    return 0;
}

你可能感兴趣的:(线段树,dfs序)