luogu_P3313 [SDOI2014]旅行

传送门

Solution

第二次学习可持久化线段树

打了一道裸题来练习一下……

• 对于每个宗教都可以开一个主席树
• 基础操作
• 树剖lca

Code 

#include<bits/stdc++.h>
#define ll long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
	return x*f;
}
#define MN 100005
#define MM 8000005
int w[MN],c[MN],N;
struct edge{int to,nex;}e[MN<<1];
int hr[MN],en;
inline void ins(int f,int t)
{
	e[++en]=(edge){t,hr[f]};hr[f]=en;
	e[++en]=(edge){f,hr[t]};hr[t]=en;
}
int fa[MN],dep[MN],mx[MN],siz[MN],pos[MN],top[MN];
void dfs1(int x,int f)
{
	dep[x]=dep[fa[x]=f]+(siz[x]=1);register int i;
	for(i=hr[x];i;i=e[i].nex) if(f^e[i].to) dfs1(e[i].to,x),siz[e[i].to]>siz[mx[x]]?mx[x]=e[i].to:0,siz[x]+=siz[e[i].to];
}
int dind=0;
void dfs2(int x,int tp)
{
	pos[x]=++dind;top[x]=tp;register int i;
	if(mx[x]) dfs2(mx[x],tp);
	for(i=hr[x];i;i=e[i].nex) if((fa[x]^e[i].to)&&(mx[x]^e[i].to)) dfs2(e[i].to,e[i].to);
}
#define mid ((l+r)>>1)
int Max[MM],Sum[MM],ls[MM],rs[MM],rt[MN],sz;
int query_m(int x,int l,int r,int a,int b)
{
	if(!x) return 0;
	if(a==l&&r==b) return Max[x];
	if(b<=mid) return query_m(ls[x],l,mid,a,b);
	if(a>mid) return query_m(rs[x],mid+1,r,a,b);
	return max(query_m(ls[x],l,mid,a,mid),query_m(rs[x],mid+1,r,mid+1,b));
}
int query_s(int x,int l,int r,int a,int b)
{
	if(!x) return 0;
	if(a==l&&r==b) return Sum[x];
	if(b<=mid) return query_s(ls[x],l,mid,a,b);
	if(a>mid) return query_s(rs[x],mid+1,r,a,b);
	return query_s(ls[x],l,mid,a,mid)+query_s(rs[x],mid+1,r,mid+1,b);
}
inline void pushup(int x){Max[x]=max(Max[ls[x]],Max[rs[x]]);Sum[x]=Sum[ls[x]]+Sum[rs[x]];}
void Modify(int &x,int l,int r,int a,int v)
{
	if(!x) x=++sz;
	if(l==r) return (void)(Max[x]=Sum[x]=v);
	if(a<=mid) Modify(ls[x],l,mid,a,v);
	else Modify(rs[x],mid+1,r,a,v);
	pushup(x);
}
int QS(int root,int x,int y)
{
	int res=0;
	while(top[x]^top[y])
	{
		if(dep[top[x]]>dep[top[y]]) res+=query_s(root,1,N,pos[top[x]],pos[x]),x=fa[top[x]];
		else res+=query_s(root,1,N,pos[top[y]],pos[y]),y=fa[top[y]];
	}
	res+=query_s(root,1,N,min(pos[x],pos[y]),max(pos[x],pos[y]));
	return res;
}
int QM(int root,int x,int y)
{
	int res=0;
	while(top[x]^top[y])
	{
		if(dep[top[x]]>dep[top[y]]) res=max(res,query_m(root,1,N,pos[top[x]],pos[x])),x=fa[top[x]];
		else res=max(res,query_m(root,1,N,pos[top[y]],pos[y])),y=fa[top[y]];
	}
	res=max(res,query_m(root,1,N,min(pos[x],pos[y]),max(pos[x],pos[y])));
	return res;
}
int main()
{
	register int i,Q,x,y;register char opt[10];
	N=read(),Q=read();
	for(i=1;i<=N;++i) w[i]=read(),c[i]=read();
	for(i=1;i<N;++i) x=read(),ins(x,read());
	dfs1(1,0);dfs2(1,1);
	for(i=1;i<=N;++i) Modify(rt[c[i]],1,N,pos[i],w[i]);
	while(Q--)
	{
		scanf("%s",opt+1);x=read();y=read();
		if(opt[1]=='C'&&opt[2]=='C') Modify(rt[c[x]],1,N,pos[x],0),Modify(rt[c[x]=y],1,N,pos[x],w[x]);
		if(opt[1]=='C'&&opt[2]=='W') Modify(rt[c[x]],1,N,pos[x],w[x]=y);
		if(opt[1]=='Q'&&opt[2]=='S') printf("%d\n",QS(rt[c[y]],x,y));
		if(opt[1]=='Q'&&opt[2]=='M') printf("%d\n",QM(rt[c[y]],x,y));
	}
	return 0;
}

---

Blog来自PaperCloud，未经允许，请勿转载，TKS！