BZOJ.4771.七彩树(可持久化线段树)

BZOJ


考虑没有深度限制,对整棵子树询问怎么做。
对于同种颜色中DFS序相邻的两个点(u,v),在(dfn[u],dfn[v])处分别(+1)(dfn[LCA(u,v)])(-1),这样答案就是求子树和了(同种颜色多余贡献的会被减掉)。
对于深度的限制,考虑维护(max{dep})棵线段树(T_i),分别表示只考虑深度在(1sim i)之间的点的贡献(下标依旧是DFS序)。因为我们发现,对于询问((x,k)),求(T_{dep[x]+k})这棵线段树中(也就是只考虑深度在(1sim dep[x]+k)的点),区间([dfn[x],ed\_dfn[x]])的和,就是答案。因为可能的问题是会统计到区间中深度(<dep[x])的点,但是区间([dfn[x],ed\_dfn[x]])中没有深度(<dep[x])的点啊!
所以我们按深度维护出可持久化线段树就OK了。
对于深度由(d-1)变成(d),考虑在(T_d)中深度为(d)的点(x)的贡献。
同样像最初提到的那样,找出与(x)颜色相同且DFS序相邻的两个点(pre,nxt),把(LCA(pre,nxt))(-1)加回去,再减掉(LCA(pre,x),LCA(x,nxt))的贡献即可(当然这两个LCA中有一个就是(LCA(pre,nxt)))。对每种颜色用(set)维护一下前驱后继。
复杂度(O(nlog n))

注意下(dep[x]+k)(max{dep})(min)

感觉代码也没有那么难写啊


//57056kb	1440ms
#include <set>
#include <cstdio>
#include <cctype>
#include <vector>
#include <cstring>
#include <algorithm>
#define gc() getchar()
#define MAXIN 300000
//#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=1e5+5;

int col[N],root[N],H[N],nxt[N],dep[N],fa[N],sz[N],son[N],top[N],Index,dfn[N],ref[N],R[N];
std::set<int> st[N];
std::vector<int> vec[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Segment_Tree
{
	#define ls son[x][0]
	#define rs son[x][1]
	#define lson ls,l,m
	#define rson rs,m+1,r
	#define S N*37//2logn!
	int tot,sum[S],son[S][2];
	#undef S
	void Modify(int &x,int y,int l,int r,int p,int v)
	{
		x=++tot, sum[x]=sum[y]+v;
		if(l==r) {ls=rs=0; return;}
		int m=l+r>>1;
		p<=m ? (rs=son[y][1],Modify(ls,son[y][0],l,m,p,v)) : (ls=son[y][0],Modify(rs,son[y][1],m+1,r,p,v));
	}
	int Query(int x,int l,int r,int L,int R)
	{
		if(!x) return 0;
		if(L<=l && r<=R) return sum[x];
		int m=l+r>>1;
		if(L<=m)
			if(m<R) return Query(lson,L,R)+Query(rson,L,R);
			else return Query(lson,L,R);
		return Query(rson,L,R);
	}
}T;

inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-48,c=gc());
	return now;
}
inline void AE(int u,int v)
{
	nxt[v]=H[u], H[u]=v;
}
inline int LCA(int u,int v)
{
	if(u==-1||v==-1) return -1;
	while(top[u]!=top[v]) dep[top[u]]>dep[top[v]]?u=fa[top[u]]:v=fa[top[v]];
	return dep[u]>dep[v]?v:u;
}
void DFS1(int x)
{
	int mx=0; sz[x]=1, son[x]=0;//!
	for(int v=H[x]; v; v=nxt[v])
		dep[v]=dep[x]+1, DFS1(v), sz[x]+=sz[v], sz[v]>mx&&(mx=sz[v],son[x]=v);
}
void DFS2(int x,int tp)
{
	top[x]=tp, ref[dfn[x]=++Index]=x, vec[dep[x]].push_back(x);
	if(son[x])
	{
		DFS2(son[x],tp);
		for(int v=H[x]; v; v=nxt[v])
			if(v!=son[x]) DFS2(v,v);
	}
	R[x]=Index;
}
void Solve()
{
	int n=read(),m=read();
	for(int i=1; i<=n; ++i) col[i]=read();
	for(int i=2; i<=n; ++i) AE(fa[i]=read(),i);
	dep[1]=1, DFS1(1), DFS2(1,1);
	int mx=1;
	for(int i=1; i<=n; ++i) mx=std::max(mx,dep[i]);
	for(int i=1; i<=mx; ++i)
	{
		root[i]=root[i-1];
		for(int j=0,l=vec[i].size(); j<l; ++j)
		{
			int x=vec[i][j],c=col[x];
			if(!st[c].empty())
			{
				std::set<int>::iterator it=st[c].upper_bound(dfn[x]);
				int nxt=it==st[c].end()?-1:ref[*it];
				int pre=it==st[c].begin()?-1:ref[*(--it)];
				int r1=LCA(pre,nxt),r2=LCA(pre,x),r3=LCA(x,nxt);
				if(r1==r2 && r3!=-1) T.Modify(root[i],root[i],1,n,dfn[r3],-1);
				else if(r1==r3 && r2!=-1) T.Modify(root[i],root[i],1,n,dfn[r2],-1);
			}
			T.Modify(root[i],root[i],1,n,dfn[x],1), st[c].insert(dfn[x]);
		}
	}
	for(int i=1,ans=0; i<=m; ++i)
	{
		int x=read()^ans,k=read()^ans;
		printf("%d
",ans=T.Query(root[std::min(mx,dep[x]+k)],1,n,dfn[x],R[x]));
	}
	T.tot=0, Index=0, memset(H,0,n+1<<2);
	for(int i=1; i<=n; ++i) std::set<int>().swap(st[i]);
	for(int i=1; i<=mx; ++i) std::vector<int>().swap(vec[i]);
}

int main()
{
	for(int T=read(); T--; Solve());
	return 0;
}
原文地址:https://www.cnblogs.com/SovietPower/p/10441572.html