bzoj 4771 七彩树

bzoj

hdu

先考虑离线做法,可以按深度离线,然后按照深度顺序加点,那么询问就可以变成在\(dep_x+d\)深度时进行子树查询.现在考虑一个点,它可以给所有它的祖先答案+1,但是如果有一个同色的点,那么在他们的\(lca\)处以及\(lca\)祖先只会有\(1\)的贡献,所以加入一个点就单点+1,加入一个同色点要把它和\(dfs\)序前面一个以及后面一个点的\(lca\)-1(还要先消掉之前前去后继\(lca\)的-1).其实就是维护每种颜色的链并.然后可以发现子树权值和就是对应答案

现在强制在线,可以先离线预处理,然后维护对应深度的可持久化线段树.每次查询找到对应最深深度的线段树直接查

#include<bits/stdc++.h>
#define LL long long
#define uLL unsigned long long
#define db double

using namespace std;
const int N=1e5+10;
int rd()
{
	int x=0,w=1;char ch=0;
	while(ch<'0'||ch>'9'){if(ch=='-') w=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
	return x*w;
}
int to[N<<1],nt[N<<1],hd[N],tot=1;
void add(int x,int y)
{
	++tot,to[tot]=y,nt[tot]=hd[x],hd[x]=tot;
	++tot,to[tot]=x,nt[tot]=hd[y],hd[y]=tot;
}
int n,m,q;
int s[N*100],ch[N*100][2],rt[N],tt;
void inst(int o1,int o2,int x,int y)
{
	s[o1]=s[o2]+y;
	int l=1,r=n;
	while(l<r)
	{
		int mid=(l+r)>>1;
		if(x<=mid)
		{
			ch[o1][0]=++tt,ch[o1][1]=ch[o2][1];
			o1=ch[o1][0],o2=ch[o2][0];
			r=mid;
		}
		else
		{
			ch[o1][0]=ch[o2][0],ch[o1][1]=++tt;
			o1=ch[o1][1],o2=ch[o2][1];
			l=mid+1;
		}
		s[o1]=s[o2]+y;
	}
}
int quer(int o,int l,int r,int ll,int rr)
{
	if(!o) return 0;
	if(ll<=l&&r<=rr) return s[o];
	int an=0,mid=(l+r)>>1;
	if(ll<=mid) an+=quer(ch[o][0],l,mid,ll,rr);
	if(rr>mid) an+=quer(ch[o][1],mid+1,r,ll,rr);
	return an;
}
int co[N],fa[N],sz[N],de[N],hs[N],top[N],dfn[N],ti;
void dfs1(int x)
{
	sz[x]=1;
	for(int i=hd[x];i;i=nt[i])
	{
		int y=to[i];
		if(y==fa[x]) continue;
		fa[y]=x,de[y]=de[x]+1,dfs1(y);
		sz[x]+=sz[y],hs[x]=sz[hs[x]]>sz[y]?hs[x]:y;
	}
}
void dfs2(int x)
{
	dfn[x]=++ti;
	if(hs[x]) top[hs[x]]=top[x],dfs2(hs[x]);
	for(int i=hd[x];i;i=nt[i])
	{
		int y=to[i];
		if(y==fa[x]||y==hs[x]) continue;
		top[y]=y,dfs2(y);
	}
}
int glca(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(de[top[x]]<de[top[y]]) swap(x,y);
		x=fa[top[x]];
	}
	return de[x]<de[y]?x:y;
}
struct node
{
	int x;
	bool operator < (const node &bb) const {return dfn[x]<dfn[bb.x];}
};
set<node> sb[N];
set<node>::iterator ft,nx;
int sq[N];
bool cmp(int aa,int bb){return de[aa]!=de[bb]?de[aa]<de[bb]:dfn[aa]<dfn[bb];}

int main()
{
//f**k
	int T=rd();
	while(T--)
	{
		n=rd(),q=rd();
		for(int i=1;i<=n;++i) fa[i]=hs[i]=top[i]=0;
		for(int i=1;i<=n;++i) sb[i].clear(),co[i]=rd();
		memset(hd,0,sizeof(int)*(n+1)),tot=1,ti=0;
		for(int i=2;i<=n;++i) add(rd(),i);
		de[1]=1,dfs1(1);
		top[1]=1,dfs2(1);
		m=0;
		for(int i=1;i<=n;++i) m=max(m,de[i]);
		for(int i=1;i<=n;++i) sq[i]=i;
		sort(sq+1,sq+n+1,cmp);
		while(tt)
		{
			s[tt]=ch[tt][0]=ch[tt][1]=0;
			--tt;
		}
		for(int i=1,j=1;i<=m;++i)
		{
			rt[i]=rt[i-1];
			while(j<=n&&de[sq[j]]==i)
			{
				int x=sq[j],las=rt[i];
				inst(rt[i]=++tt,las,dfn[x],1);
				if(!sb[co[x]].empty())
				{
					nx=sb[co[x]].upper_bound((node){x});
					if(nx!=sb[co[x]].begin())
					{
						ft=--nx;
						++nx;
						las=rt[i],inst(rt[i]=++tt,las,dfn[glca(x,(*ft).x)],-1);
						if(nx!=sb[co[x]].end()) las=rt[i],inst(rt[i]=++tt,las,dfn[glca((*ft).x,(*nx).x)],1);
					}
					if(nx!=sb[co[x]].end())
					{
						las=rt[i],inst(rt[i]=++tt,las,dfn[glca(x,(*nx).x)],-1);
					}
				}
				sb[co[x]].insert((node){x});
				++j;
			}
		}
		int ans=0;
		while(q--)
		{
			int x=rd()^ans,d=min((rd()^ans)+de[x],m);
			printf("%d\n",(ans=quer(rt[d],1,n,dfn[x],dfn[x]+sz[x]-1)));
		}
	}
	return 0; 
}
posted @ 2019-09-16 23:00  ✡smy✡  阅读(108)  评论(0编辑  收藏  举报