传送门
题解:
首先考虑如果没有距离限制,显然可以考虑树上差分。
讲相同颜色的点拿出来,按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;
}