BZOJ.3531.旅行(树链剖分 动态开点)

题目链接

无优化版本(170行)：

/*
首先树剖可以维护树上的链Sum、Max
可以对每个宗教建一棵线段树，那这题就很好做了
不过10^5需要动态开点 (不明白为什么nlogn不需要回收就可以 不是每个Insert加log个节点?)
操作修改完更改原数列！盲人。。
*/
#include<cstdio>
#include<cctype>
#include<algorithm>
#define gc() getchar()
#define now node[rt]
#define lson node[rt].ls
#define rson node[rt].rs
const int N=1e5+5,MAXN=N*19;

int n,q,cnt,H[N],Enum,to[N<<1],nxt[N<<1],fa[N],dep[N],id[N],son[N],sz[N],top[N];
int idx,pos[N],W[N],rel[N],root[N],pool[MAXN];
struct Seg_Tree
{
	struct Node
	{
		int sum,maxn,ls,rs,l,r;
		inline void Init() {sum=maxn=l=r=ls=rs=0;}
	}node[MAXN];
	inline int new_Node() {return pool[++idx];}
	inline void del_Node(int rt) {now.Init(), pool[idx--]=rt;}
	inline void PushUp(int rt)
	{
		now.sum = node[lson].sum + node[rson].sum,
		now.maxn= std::max(node[lson].maxn, node[rson].maxn);
	}
	void Insert(int l,int r,int &rt,int x)
	{
		if(!rt) rt=new_Node(), now.l=l, now.r=r;
		if(l==r) now.sum=now.maxn=W[x];
		else
		{
			int m=l+r>>1;
			if(id[x]<=m) /*lson?0:lson=new_Node(),*/ Insert(l,m,lson,x);//不想打括号--
			else /*rson?0:rson=new_Node(),*/ Insert(m+1,r,rson,x);
			PushUp(rt);
		}
	}
	void Delete(int rt,int x,int &son)//可以用Insert将其值变为0来删除 但是就不能重复利用了
	{
		if(now.l==now.r) son=0, del_Node(rt);//父亲的这个儿子也要删!
		else
		{
			int m=now.l+now.r>>1;
			if(id[x]<=m) Delete(lson,x,lson);
			else Delete(rson,x,rson);
			if(now.sum==W[x]) del_Node(rt), son=0;
			else PushUp(rt);
		}
	}
	void Modify(int rt,int p,int v)
	{
		if(now.l==now.r) now.maxn=now.sum=v;
		else
		{
			int m=now.l+now.r>>1;
			if(p<=m) Modify(lson,p,v);
			else Modify(rson,p,v);
			PushUp(rt);
		}
	}
	int Query_Max(int rt,int L,int R)
	{
		if(!rt) return 0;
		if(L<=now.l && now.r<=R) return now.maxn;
		int m=now.l+now.r>>1;
		if(L<=m)
			if(m<R) return std::max(Query_Max(lson,L,R),Query_Max(rson,L,R));
			else return Query_Max(lson,L,R);
		return Query_Max(rson,L,R);
	}
	int Query_Sum(int rt,int L,int R)
	{
		if(!rt) return 0;
		if(L<=now.l && now.r<=R) return now.sum;
		int m=now.l+now.r>>1;
		if(L<=m)
			if(m<R) return Query_Sum(lson,L,R)+Query_Sum(rson,L,R);
			else return Query_Sum(lson,L,R);
		return Query_Sum(rson,L,R);
	}
}t;
#undef now
inline int read()
{
	int now=0,f=1;register char c=gc();
	for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now*f;
}
inline void AddEdge(int u,int v)
{
	to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
	to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
}
void DFS1(int x)
{
	sz[x]=1; int mx=0;
	for(int v,i=H[x]; i; i=nxt[i])
		if((v=to[i])!=fa[x])
		{
			fa[v]=x, dep[v]=dep[x]+1, DFS1(v), sz[x]+=sz[v];
			if(mx<sz[v]) mx=sz[v], son[x]=v;
		}
}
void DFS2(int x,int tp)
{
	id[x]=++cnt, top[x]=tp;
	if(son[x])
	{
		DFS2(son[x],tp);
		for(int i=H[x]; i; i=nxt[i])
			if(to[i]!=fa[x] && to[i]!=son[x])
				DFS2(to[i], to[i]);
	}
}
void Query_Chain_Sum(int x,int y)
{
	long long res=0;int rt=root[rel[x]];
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) std::swap(x,y);
		res += t.Query_Sum(rt,id[top[x]],id[x]);
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) std::swap(x,y);
	res += t.Query_Sum(rt,id[x],id[y]);
	printf("%lld\n",res);
}
void Query_Chain_Max(int x,int y)
{
	int res=0, rt=root[rel[x]];
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) std::swap(x,y);
		res = std::max(res, t.Query_Max(rt,id[top[x]],id[x]));
		x=fa[top[x]];
	}
	if(dep[x]>dep[y]) std::swap(x,y);
	res = std::max(res, t.Query_Max(rt,id[x],id[y]));
	printf("%d\n",res);
}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("3531.in","r",stdin);
	freopen("my.out","w",stdout);
#endif

	n=read(),q=read();
	for(int i=1; i<MAXN; ++i) pool[i]=i;
	for(int i=1; i<=n; ++i) W[i]=read(), rel[i]=read();
	for(int u,v,i=1; i<n; ++i) u=read(),v=read(),AddEdge(u,v);
	DFS1(1), DFS2(1,1);
	for(int i=1; i<=n; ++i) t.Insert(1,n,root[rel[i]],i);
	char opt[5]; int x,y;
	while(q--)
	{
		scanf("%s",opt), x=read(), y=read();
		switch(opt[1])
		{
			case 'C': t.Delete(root[rel[x]],x,root[rel[x]]), rel[x]=y, t.Insert(1,n,root[y],x);
					  break;
			case 'W': t.Modify(root[rel[x]],id[x],y), W[x]=y;//!
					  break;
			case 'S': Query_Chain_Sum(x,y);
					  break;
			case 'M': Query_Chain_Max(x,y);
					  break;
		}
	}
	return 0;
}

优化(短多了)：

//不知道为什么只需nlogn个节点 不需要回收
//+各种函数简化
#include<cstdio>
#include<cctype>
#include<algorithm>
#define gc() getchar()
#define now node[rt]
#define lson node[rt].ls
#define rson node[rt].rs
const int N=1e5+5,MAXN=N*19;

int n,q,cnt,H[N],Enum,to[N<<1],nxt[N<<1],fa[N],dep[N],id[N],son[N],sz[N],top[N];
int idx,pos[N],W[N],rel[N],root[N],pool[MAXN];
struct Seg_Tree
{
	struct Node
	{
		int sum,maxn,ls,rs;
		inline void Init() {sum=maxn=ls=rs=0;}
	}node[MAXN];
	inline int new_Node() {return ++idx;}
//	inline int new_Node() {return pool[++idx];}
//	inline void del_Node(int rt) {now.Init(), pool[idx--]=rt;}
	inline void PushUp(int rt)
	{
		now.sum = node[lson].sum + node[rson].sum,
		now.maxn= std::max(node[lson].maxn, node[rson].maxn);
	}
	void Update(int l,int r,int &rt,int p,int v)
	{
		if(!rt) rt=new_Node();
		if(l==r) now.sum=now.maxn=v;
		else
		{
			int m=l+r>>1;
			if(p<=m) Update(l,m,lson,p,v);
			else Update(m+1,r,rson,p,v);
			PushUp(rt);
		}
	}
	int Query(int l,int r,int rt,int L,int R,bool opt)
	{
		if(!rt) return 0;
		if(L<=l && r<=R) return opt?now.sum:now.maxn;
		int m=l+r>>1;
		if(L<=m)
			if(m<R) return opt?Query(l,m,lson,L,R,opt)+Query(m+1,r,rson,L,R,opt)
							  :std::max(Query(l,m,lson,L,R,opt),Query(m+1,r,rson,L,R,opt));
			else return Query(l,m,lson,L,R,opt);
		return Query(m+1,r,rson,L,R,opt);
	}
}t;
#undef now
inline int read()
{
	int now=0,f=1;register char c=gc();
	for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
	for(;isdigit(c);now=now*10+c-'0',c=gc());
	return now*f;
}
inline void AddEdge(int u,int v)
{
	to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
	to[++Enum]=u, nxt[Enum]=H[v], H[v]=Enum;
}
void DFS1(int x)
{
	sz[x]=1; int mx=-1;
	for(int v,i=H[x]; i; i=nxt[i])
		if((v=to[i])!=fa[x])
		{
			fa[v]=x, dep[v]=dep[x]+1, DFS1(v), sz[x]+=sz[v];
			if(mx<sz[v]) mx=sz[v], son[x]=v;
		}
}
void DFS2(int x,int tp)
{
	id[x]=++cnt, top[x]=tp;
	if(son[x])
	{
		DFS2(son[x],tp);
		for(int i=H[x]; i; i=nxt[i])
			if(to[i]!=fa[x] && to[i]!=son[x])
				DFS2(to[i], to[i]);
	}
}
void Query_Chain(int x,int y,bool opt)
{
	int res=0;int rt=root[rel[x]];
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) std::swap(x,y);
		if(opt) res += t.Query(1,n,rt,id[top[x]],id[x],1);
		else res = std::max(res, t.Query(1,n,rt,id[top[x]],id[x],0));
		x=fa[top[x]];
	}
	if(id[x]>id[y]) std::swap(x,y);
	if(opt) res += t.Query(1,n,rt,id[x],id[y],1);
	else res = std::max(res, t.Query(1,n,rt,id[x],id[y],0));
	printf("%d\n",res);
}

int main()
{
#ifndef ONLINE_JUDGE
	freopen("35312.in","r",stdin);
	freopen("my.out","w",stdout);
#endif

	n=read(),q=read();
	for(int i=1; i<MAXN; ++i) pool[i]=i;
	for(int i=1; i<=n; ++i) W[i]=read(), rel[i]=read();
	for(int u,v,i=1; i<n; ++i) u=read(),v=read(),AddEdge(u,v);
	DFS1(1), DFS2(1,1);
	for(int i=1; i<=n; ++i) t.Update(1,n,root[rel[i]],id[i],W[i]);
	char opt[5]; int x,y;
	while(q--)
	{
		scanf("%s",opt), x=read(), y=read();
		switch(opt[1])
		{
			case 'C': t.Update(1,n,root[rel[x]],id[x],0), rel[x]=y, t.Update(1,n,root[y],id[x],W[x]);
					  break;
			case 'W': t.Update(1,n,root[rel[x]],id[x],y),W[x]=y/*!*/; break;
			case 'S': Query_Chain(x,y,1); break;
			case 'M': Query_Chain(x,y,0); break;
		}
	}
	return 0;
}