【BZOJ4771】七彩树（可持久化线段树）（树上差分）

传送门

---

题解：

首先考虑如果没有距离限制，显然可以考虑树上差分。

讲相同颜色的点拿出来，按dfs序排序，每个点的位置+1，相邻两点LCA处-1。

询问就是子树标记之和。

但是现在需要考虑深度，我们把点先按照深度为第一关键字，dfs序为第二关键字排个序。

然后从深度浅的逐渐加点，每个线段树从上一个继承所有标记，然后加入新店，按照dfs维护一下差分标记。

注意到我们需要动态维护当前已经加点的dfs序和当前加点的前驱后继，每个颜色开一个set即可。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<22|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	
	template<typename T>
	inline T get(){
		char c;T num;
		while(!isdigit(c=gc()));num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
	inline int gi(){return get<int>();}
}
using namespace IO;

using std::cerr;
using std::cout;

cs int N=1e5+7;

int n,m;

namespace SGT{
	cs int N=::N*60;
	int lc[N],rc[N],siz[N],tot;
	inline void clear(){tot=0;}
	inline void ins(int &u,int l,int r,int p,int t){
		int v=++tot;siz[v]=siz[u]+t;
		lc[v]=lc[u],rc[v]=rc[u];u=v;
		if(l==r)return ;int mid=l+r>>1;
		(p<=mid)?ins(lc[u],l,mid,p,t):ins(rc[u],mid+1,r,p,t);
	}
	inline int query(int u,int l,int r,int ql,int qr){
		if(ql<=l&&r<=qr)return siz[u];
		int mid=l+r>>1;
		if(qr<=mid)return query(lc[u],l,mid,ql,qr);
		if(mid<ql)return query(rc[u],mid+1,r,ql,qr);
		return query(lc[u],l,mid,ql,qr)+query(rc[u],mid+1,r,ql,qr);
	}
}
int rt[N];

int last[N],nxt[N],to[N],ecnt;
inline void adde(int u,int v){
	nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v;
}

int fa[N],d[N];
int siz[N],son[N],top[N];
int in[N],out[N],pos[N],clk;
std::vector<int> rec[N];
int mxd,nd[N],ct;

void dfs1(int u,int p){
	pos[in[u]=++clk]=u;siz[u]=1;
	d[u]=d[p]+1;rec[d[u]].push_back(u);
	for(int re e=last[u],v;v=to[e];e=nxt[e]){
		dfs1(v,u);siz[u]+=siz[v];if(siz[v]>siz[son[u]])son[u]=v;
	}
	out[u]=clk;
}
void dfs2(int u,int tp){
	top[u]=tp;
	for(int re e=last[u],v;v=to[e];e=nxt[e])
	dfs2(v,v==son[u]?tp:v);
}
inline int LCA(int u,int v){
	while(top[u]!=top[v])d[top[u]]<d[top[v]]?v=fa[top[v]]:u=fa[top[u]];
	return d[u]<d[v]?u:v;
}

int c[N];
std::set<int> s[N];
typedef std::set<int>::iterator iter;

inline void solve(){
	n=gi(),m=gi();
	for(int re i=1;i<=n;++i)c[i]=gi();
	for(int re i=2;i<=n;++i)adde(fa[i]=gi(),i);
	dfs1(1,0),dfs2(1,1);ct=0;
	for(int re i=1;;++i){
		if(!rec[i].size()){mxd=i-1;break;}
		for(int re j=0;j<rec[i].size();++j)nd[++ct]=rec[i][j];
	}
	for(int re i=1;i<=n;++i){
		if(d[nd[i]]!=d[nd[i-1]])rt[d[nd[i]]]=rt[d[nd[i-1]]];
		int u=nd[i],&rt=::rt[d[u]],pre=0,suc=0;
		SGT::ins(rt,1,n,in[u],1);
		if(!s[c[u]].size()){s[c[u]].insert(in[u]);continue;}
		iter p=s[c[u]].lower_bound(in[u]);
		if(p!=s[c[u]].end()){
			suc=pos[*p];
			SGT::ins(rt,1,n,in[LCA(u,suc)],-1);
		}
		if(p!=s[c[u]].begin()){
			pre=pos[*--p];
			SGT::ins(rt,1,n,in[LCA(u,pre)],-1);
		}
		if(pre&&suc)SGT::ins(rt,1,n,in[LCA(pre,suc)],1);
		s[c[u]].insert(in[u]);
	}int ans=0;
	while(m--){
		int u=gi()^ans,dd=gi()^ans;
		dd=std::min(d[u]+dd,mxd);
		cout<<(ans=SGT::query(rt[dd],1,n,in[u],out[u]))<<"\n";
	}
	SGT::clear();
	memset(last,0,sizeof(int)*(n+1));ecnt=0;
	memset(son,0,sizeof(int)*(n+1));clk=0;
	for(int re i=1;i<=n;++i)s[i].clear();
	for(int re i=1;i<=mxd;++i)rec[i].clear();mxd=0;
}

signed main(){
#ifdef zxyoi
	freopen("colorful.in","r",stdin);
#endif
	int T=gi();
	while(T--)solve();
	return 0;
}