2018蓝桥杯模拟赛（一）--青出于蓝而胜于蓝（线段树）

题目链接：https://nanti.jisuanke.com/t/20690

样例输入

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

题意：给出一个根节点，然后判断每一个节点的孩子节点小于此节点的个数。

思路：将树的前序遍历跑出来，每个节点和它的孩子节点都在一个区间里边，然后相当于在一个数组区间中查询小于某一个值的数的个数，我们维护一个最大值，最小值的线段树，如果最大值小于x，那个数就是这个区间的长度；如果最小值小于等于x，那么为0;

PS:当初不会做，想看答疑，发现蓝桥杯的答疑还要收费10元，坑，然后跟队友讨论出来了。

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
typedef long long ll;
#define clr(x,y) memset(x,y,sizeof x)
#define INF 0x3f3f3f3f

vector<int>g[maxn];
int ans[maxn];
int maxs[maxn << 2],mins[maxn <<2];

int a[maxn];
int L[maxn],R[maxn];
int dfs(int u,int pre,int &len)
{
    a[len] = u;
    L[u] = len;
    for(int i = 0;i < g[u].size();i ++)
    {
        int v = g[u][i];
        if(v == pre)continue;
        dfs(v,u,++ len);
    }
    R[u] = len;

}
void build(int l,int r,int rt)
{
    if(l == r)
    {
        mins[rt] = maxs[rt] = a[l];return ;
    }
    int mid = (l + r) >> 1;
    build(l,mid,rt <<1);build(mid + 1,r,rt << 1|1);
    maxs[rt] = max(maxs[rt << 1],maxs[rt << 1|1]);
    mins[rt] = min(mins[rt << 1],mins[rt << 1|1]);
}
int fun(int l,int r,int x,int rt)
{
    if(l == r)
    {
        return a[l] < x ? 1 : 0;
    }
    if(mins[rt] >= x)
        return 0;
    if(maxs[rt] < x)
        return (r - l + 1);
    int mid = (l + r) >> 1;
    int ret = 0;
    if(mins[rt << 1] < x)
        ret += fun(l,mid,x,rt << 1);
    if(mins[rt << 1|1] < x)
        ret += fun(mid + 1,r,x,rt << 1|1);
    return ret;
}
int query(int L,int R,int x,int l,int r,int rt)
{
    if(L <= l && R >= r)
    {
        if(mins[rt] >= x)
            return 0;
        if(maxs[rt] < x)
            return (r - l + 1);
        return fun(l,r,x,rt);
    }
    int mid = (l + r) >> 1;
    int ret = 0;
    if(L <= mid)
        ret += query(L,R,x,l,mid,rt << 1);
    if(R >= mid + 1)
        ret += query(L,R,x,mid + 1,r,rt << 1|1);
    return ret;
}

int main()
{
    int n,m;
    while( ~ scanf("%d%d",&n,&m))
    {
        for(int i = 0;i <= n;i ++)g[i].clear();
        for(int i = 1;i <= n - 1;i ++)
        {
            int x,y;scanf("%d%d",&x,&y);
            g[x].push_back(y);g[y].push_back(x);
        }
        int len = 1;
        dfs(m,m,len);
        build(1,n,1);

//        for(int i = 1;i <= n;i ++)
//            cout <<a[i] << " " << L[a[i]] << " " << R[a[i]] << endl;
//        for(int i = 1;i <= n;i ++)
//            cout << a[i] << " ";puts("");
        for(int i = 1;i <= n;i ++)
        {
            ans[i] = query(L[i],R[i],i,1,n,1);
        }
        for(int i = 1;i <= n;i ++)
            printf("%d%c",ans[i],i < n ? ' ' : '\n');
    }
    return 0;
}