bzoj P3531 [Sdoi2014]旅行-优快云博客

这道题学到了动态开空间线段树的写法，感觉跟主席树有一定的类似。

说一下怎么动态开吧。

首先回忆一下线段树的过程，我们发现对于这题，每一颗线段树都有很多的子结点是0，也就是说这些节点都是没用的，所以我们考虑对于每一个非0叶子结点到根都是非0的，但是其他都是为0的，都是没有用的，那么为何要为它们开内存存储呢，所以我们换种方式存储对每一个节点都有一个编号，然后结构体内的l，r都是存的l，r的编号，对于没用的节点都不开内存给它。

再说下这题的题解，开c个线段树来维护c个宗教，所以每一宗教就直接树链剖分，然后动态开内存空间要开大点。

代码：

#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
const int Maxn=1000005;
struct node{int v,next;}Qt[Maxn*2];
struct segment{int l,r,mx,sum;}t[Maxn*4];
int n,m,cnt,tmp,SIZE;
int w[Maxn],c[Maxn],head[Maxn],size[Maxn],fa[Maxn],depth[Maxn],pos[Maxn],top[Maxn],root[Maxn];
inline void swap(int &a,int &b){int t=a;a=b;b=t;}
inline int max(int a,int b){return a>b?a:b;}
inline void Insert(int u,int v)
{
	cnt++;
	Qt[cnt].v=v;
	Qt[cnt].next=head[u];
	head[u]=cnt;
}
void dfs1(int x)
{
	size[x]=1;
	for(int i=head[x];i;i=Qt[i].next)
	{
		if(Qt[i].v==fa[x])continue;
		depth[Qt[i].v]=depth[x]+1;
		fa[Qt[i].v]=x;
		dfs1(Qt[i].v);
		size[x]+=size[Qt[i].v];
	}
}
void dfs2(int x,int tp)
{
	int k=0;
	pos[x]=++tmp;
	top[x]=tp;
	for(int i=head[x];i;i=Qt[i].next)
		if(depth[Qt[i].v]>depth[x]&&size[Qt[i].v]>size[k])
			k=Qt[i].v;
	if(k)
		dfs2(k,tp);
	for(int i=head[x];i;i=Qt[i].next)
		if(depth[Qt[i].v]>depth[x]&&Qt[i].v!=k)
			dfs2(Qt[i].v,Qt[i].v);
}
inline void change(int &k,int l,int r,int x,int d)
{
	if(!k)k=++SIZE;
	if(l==r){t[k].mx=t[k].sum=d;return ;}
	int mid=(l+r)>>1;
	if(x<=mid)change(t[k].l,l,mid,x,d);
	else change(t[k].r,mid+1,r,x,d);
	t[k].mx=max(t[t[k].l].mx,t[t[k].r].mx);
	t[k].sum=t[t[k].l].sum+t[t[k].r].sum;
}
inline int Qmax(int k,int l,int r,int lf,int rf)
{
	if(l>=lf&&r<=rf)return t[k].mx;
	int mid=(l+r)>>1,mx=0;
	if(mid>=lf)mx=max(mx,Qmax(t[k].l,l,mid,lf,rf));
	if(mid<rf)mx=max(mx,Qmax(t[k].r,mid+1,r,lf,rf));
	return mx;
}
inline int Qsum(int k,int l,int r,int lf,int rf)
{
	if(l>=lf&&r<=rf)return t[k].sum;
	int mid=(l+r)>>1,sum=0;
	if(mid>=lf)sum+=Qsum(t[k].l,l,mid,lf,rf);
	if(mid<rf)sum+=Qsum(t[k].r,mid+1,r,lf,rf);
	return sum;
}
inline int solvesum(int _c,int x,int y)
{
	int sum=0;
	while(top[x]!=top[y])
	{
		if(depth[top[x]]<depth[top[y]])swap(x,y);
		sum+=Qsum(root[_c],1,n,pos[top[x]],pos[x]);
		x=fa[top[x]];
	}
	if(pos[x]>pos[y])swap(x,y);
	sum+=Qsum(root[_c],1,n,pos[x],pos[y]);
    return sum;
}
inline int solvemax(int _c,int x,int y)
{
	int mx=0;
	while(top[x]!=top[y])
	{
		if(depth[top[x]]<depth[top[y]])swap(x,y);
		mx=max(mx,Qmax(root[_c],1,n,pos[top[x]],pos[x]));
		x=fa[top[x]];
	}
	if(pos[x]>pos[y])swap(x,y);
	mx=max(mx,Qmax(root[_c],1,n,pos[x],pos[y]));
    return mx;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		scanf("%d%d",&w[i],&c[i]);
	for(int i=1;i<=n-1;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		Insert(x,y);
		Insert(y,x);
	}
	dfs1(1);
	dfs2(1,1);
	for(int i=1;i<=n;i++)
		change(root[c[i]],1,n,pos[i],w[i]);
	scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        char ch[5];
		int x,y;
        scanf("%s",ch);
        scanf("%d%d",&x,&y);
        if(ch[1]=='C')
        	change(root[c[x]],1,n,pos[x],0),c[x]=y,change(root[c[x]],1,n,pos[x],w[x]);
        else if(ch[1]=='W')
			change(root[c[x]],1,n,pos[x],y),w[x]=y;
        else if(ch[1]=='S')
    		printf("%d\n",solvesum(c[x],x,y));
        else if(ch[1]=='M')
        	printf("%d\n",solvemax(c[x],x,y));
    }
	return 0;
}