首页 > 代码库 > BZOJ 4756 线段树合并(线段树)
BZOJ 4756 线段树合并(线段树)
思路:
1.最裸的线段树合并
2.
我们可以观察到子树求一个东西
那我们直接DFS序好了
入队的时候统计一下有多少比他大的
出的时候统计一下
减一下
搞定~
线段树合并代码:
//By SiriusRen#include <cstdio>#include <cstring>#include <algorithm>using namespace std;const int N=100050;int n,col[N],cpy[N],tree[N*100],lson[N*100],rson[N*100],cnt,root[N];int first[N],next[N],v[N],tot,ans[N],fa[N];void add(int x,int y){v[tot]=y,next[tot]=first[x],first[x]=tot++;}void insert(int l,int r,int &pos,int num,int wei){ pos=++cnt,tree[pos]+=wei; if(l==r)return; int mid=(l+r)>>1; if(mid<num)insert(mid+1,r,rson[pos],num,wei); else insert(l,mid,lson[pos],num,wei);}int merge(int now,int last){ if(!now||!last)return now^last; lson[now]=merge(lson[now],lson[last]); rson[now]=merge(rson[now],rson[last]); tree[now]=tree[lson[now]]+tree[rson[now]]; return now;}int query(int l,int r,int pos,int L,int R){ if(l>=L&&r<=R)return tree[pos]; int mid=(l+r)>>1; if(mid<L)return query(mid+1,r,rson[pos],L,R); else if(mid>=R)return query(l,mid,lson[pos],L,R); else return query(l,mid,lson[pos],L,R)+query(mid+1,r,rson[pos],L,R);}void dfs(int x){ insert(0,n,root[x],col[x],1); for(int i=first[x];~i;i=next[i])if(v[i]!=fa[x])dfs(v[i]); for(int i=first[x];~i;i=next[i])if(v[i]!=fa[x])root[x]=merge(root[x],root[v[i]]); ans[x]=query(0,n,root[x],col[x]+1,n);}int main(){ memset(first,-1,sizeof(first)); scanf("%d",&n); for(int i=1;i<=n;i++)scanf("%d",&col[i]),cpy[i]=col[i]; sort(cpy+1,cpy+1+n); for(int i=1;i<=n;i++)col[i]=lower_bound(cpy+1,cpy+1+n,col[i])-cpy-1; for(int i=2;i<=n;i++)scanf("%d",&fa[i]),add(fa[i],i); dfs(1); for(int i=1;i<=n;i++)printf("%d\n",ans[i]);}
BZOJ 4756 线段树合并(线段树)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。