BZOJ - 4771 七彩树 (可持久化线段树合并)

题目链接

对每个结点建立两棵线段树，一棵记录该结点的子树下每种颜色对应的最小深度，另一棵记录子树下的每个深度有多少结点（每种颜色的结点只保留最浅的深度即可），自底而上令父节点继承子结点的线段树，如果合并两棵颜色线段树时发现某种颜色重复，则在深度线段树上把较深的深度对应的位置-1。

注意由于强制在线，深度线段树的合并以及更新都需要可持久化。

(ps:不能用map代替颜色线段树，会TLE~~)

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 typedef double db;
 const int N=1e5+;
 int n,m,hd[N],ne,rt1[N],rt2[N],tot1,tot2,a[N],dep[N];
 struct E {int v,nxt;} e[N];
 struct D1 {int ls,rs,x;} tr1[N*];
 struct D2 {int ls,rs,x;} tr2[N*];
 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}
 #define mid ((l+r)>>1)
 int newnode1() {tr1[++tot1]= {,,}; return tot1;}
 int newnode2() {tr2[++tot2]= {,,}; return tot2;}
 void upd1(int& u,int p,int x,int l=,int r=n) {
     if(!u)u=newnode1();
     if(l==r) {tr1[u].x=x; return;}
     p<=mid?upd1(tr1[u].ls,p,x,l,mid):upd1(tr1[u].rs,p,x,mid+,r);
 }
 void upd2(int& w,int u,int p,int x,int l=,int r=n) {
     w=newnode2();
     tr2[w].x=tr2[u].x+x;
     if(l==r)return;
     if(p<=mid)upd2(tr2[w].ls,tr2[u].ls,p,x,l,mid),tr2[w].rs=tr2[u].rs;
     else upd2(tr2[w].rs,tr2[u].rs,p,x,mid+,r),tr2[w].ls=tr2[u].ls;
 }
 void mg1(int uu,int& u,int v,int l=,int r=n) {
     if(!u) {u=v; return;}
     if(!v)return;
     if(l==r) {
         if(!tr1[u].x)tr1[u].x=tr1[v].x;
         else if(!tr1[v].x);
         else {
             int mx=max(tr1[u].x,tr1[v].x),mi=min(tr1[u].x,tr1[v].x);
             upd2(rt2[uu],rt2[uu],mx,-),tr1[u].x=mi;
         }
         return;
     }
     mg1(uu,tr1[u].ls,tr1[v].ls,l,mid);
     mg1(uu,tr1[u].rs,tr1[v].rs,mid+,r);
 }
 void mg2(int& w,int u,int v,int l=,int r=n) {
     if(!u) {w=v; return;}
     if(!v) {w=u; return;}
     w=newnode2();
     tr2[w].x=tr2[u].x+tr2[v].x;
     if(l==r)return;
     mg2(tr2[w].ls,tr2[u].ls,tr2[v].ls,l,mid);
     mg2(tr2[w].rs,tr2[u].rs,tr2[v].rs,mid+,r);
 }
 void dfs(int u,int d) {
     rt1[u]=rt2[u]=,dep[u]=d;
     upd1(rt1[u],a[u],dep[u]),upd2(rt2[u],rt2[u],dep[u],);
     for(int i=hd[u]; ~i; i=e[i].nxt) {
         int v=e[i].v;
         dfs(v,d+);
         mg1(u,rt1[u],rt1[v]);
         mg2(rt2[u],rt2[u],rt2[v]);
     }
 }
 int qry(int u,int L,int R,int l=,int r=n) {
     if(l>=L&&r<=R)return tr2[u].x;
     if(l>R||r<L)return ;
     return qry(tr2[u].ls,L,R,l,mid)+qry(tr2[u].rs,L,R,mid+,r);
 }
 int main() {
     int T;
     for(scanf("%d",&T); T--;) {
         memset(hd,-,sizeof hd),ne=tot1=tot2=;
         scanf("%d%d",&n,&m);
         for(int i=; i<=n; ++i)scanf("%d",&a[i]);
         for(int i=; i<=n; ++i) {
             int f;
             scanf("%d",&f);
             addedge(f,i);
         }
         dfs(,);
         for(int ans=; m--;) {
             int u,d;
             scanf("%d%d",&u,&d),u^=ans,d^=ans;
             printf("%d\n",ans=qry(rt2[u],,min(dep[u]+d,n)));
         }
     }
     return ;
 }