BZOJ 4756 [Usaco2017 Jan]Promotion Counting 线段树合并

题目大意:给出一棵有根树,每个点有一个权值,问对于每一个点的子树中比这个点大的点的个数。

首先权值很大需要离散化。(2017-7-13UPD:不用离散化也可反正也是动态开点
对于每一个点建一个权值线段树。遍历每一棵子树,然后将子树的线段树合并到自己的线段树上,在查询之后将自己插入。时间复杂度O(nlogn)

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 100005
using namespace std;
struct Node {
    Node *ch[2];
    int l,r,sum;
    Node(int _l,int _r):l(_l),r(_r) {
        sum=0;
        ch[0]=ch[1]=NULL;
    }
    Node() {}
    void maintain() {
        sum=0;
        if(ch[0]) sum+=ch[0]->sum;
        if(ch[1]) sum+=ch[1]->sum;
        return ;
    }
    void* operator new (size_t) {
        static Node *mempool,*C;
        if(mempool==C) mempool=(C=new Node[1<<20])+(1<<20);
        return C++;
    }
}*root[N];
struct Edge {
    int to,nxt;
    Edge() {}
    Edge(int _to,int _nxt):to(_to),nxt(_nxt) {}
}e[N];
int n,m,tot=-1,fir[N],a[N],b[N],ans[N];
void Add_Edge(int x,int y) {
    e[++tot]=Edge(y,fir[x]), fir[x]=tot;
    return ;
}
void Merge(Node*& x,Node*& y) {
    if(!y) return ;
    if(!x) {
        x=y;
        return ;
    }
    x->sum+=y->sum;
    Merge(x->ch[0],y->ch[0]), Merge(x->ch[1],y->ch[1]);
    return ;
}
void Insert(Node*& o,int v,int L,int R) {
    if(!o) o=new Node(L,R);
    if(o->l==o->r) {
        o->sum++;
        return ;
    }
    int mid=L+R>>1;
    if(v<=mid) Insert(o->ch[0],v,L,mid);
    else Insert(o->ch[1],v,mid+1,R);
    o->maintain();
    return ;
}
int Query(Node* o,int v) {
    if(!o) return 0;
    if(v==o->r) return o->sum;
    int mid=o->l+o->r>>1;
    if(v<=mid) return Query(o->ch[0],v);
    return (o->ch[0] ? o->ch[0]->sum : 0)+Query(o->ch[1],v);
}
void dfs(int x) {
    for(int i=fir[x];~i;i=e[i].nxt) {
        dfs(e[i].to);
        Merge(root[x],root[e[i].to]);
    }
    ans[x]=Query(root[x],n-a[x]);
    Insert(root[x],n-a[x]+1,1,n);
    return ;
}
int main() {
    memset(fir,-1,sizeof fir);
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",a+i), b[i]=a[i];
    sort(b+1,b+n+1);
    m=unique(b+1,b+n+1)-b-1;
    for(int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+1+m,a[i])-b;
    for(int i=2;i<=n;i++) {
        int x;
        scanf("%d",&x);
        Add_Edge(x,i);
    }
    dfs(1);
    for(int i=1;i<=n;i++) printf("%d\n",ans[i]);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值