BZOJ3531: [Sdoi2014]旅行

Description

 S国有N个城市，编号从1到N。城市间用N-1条双向道路连接，满足
从一个城市出发可以到达其它所有城市。每个城市信仰不同的宗教，如飞天面条神教、隐形独角兽教、绝地教都是常见的信仰。为了方便，我们用不同的正整数代表各种宗教，  S国的居民常常旅行。旅行时他们总会走最短路，并且为了避免麻烦，只在信仰和他们相同的城市留宿。当然旅程的终点也是信仰与他相同的城市。S国政府为每个城市标定了不同的旅行评级，旅行者们常会记下途中（包括起点和终点）留宿过的城市的评级总和或最大值。
    在S国的历史上常会发生以下几种事件：
”CC x c”：城市x的居民全体改信了c教；
”CW x w”：城市x的评级调整为w;
”QS x y”：一位旅行者从城市x出发，到城市y，并记下了途中留宿过的城市的评级总和；
”QM x y”：一位旅行者从城市x出发，到城市y，并记下了途中留宿过
的城市的评级最大值。
    由于年代久远，旅行者记下的数字已经遗失了，但记录开始之前每座城市的信仰与评级，还有事件记录本身是完好的。请根据这些信息，还原旅行者记下的数字。    为了方便，我们认为事件之间的间隔足够长，以致在任意一次旅行中，所有城市的评级和信仰保持不变。

Input

    输入的第一行包含整数N，Q依次表示城市数和事件数。
    接下来N行，第i+l行两个整数Wi，Ci依次表示记录开始之前，城市i的
评级和信仰。
    接下来N-1行每行两个整数x，y表示一条双向道路。
    接下来Q行，每行一个操作，格式如上所述。

Output

    对每个QS和QM事件，输出一行，表示旅行者记下的数字。

Sample Input

5 6
3 1
2 3
1 2
3 3
5 1
1 2
1 3
3 4
3 5
QS 1 5
CC 3 1
QS 1 5
CW 3 3
QS 1 5
QM 2 4

Sample Output

8
9
11

3

题解：

树剖+线段树

用树剖对于每一个宗教建立一棵线段树，然后维护。

主要是打代码。。。。。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int N=100005;
const int M=10000005;
int n,q;
int c[N],w[N];
struct node{
	int x,y,next;
}sa[N*2];int len=0,first[N*2];
struct node2{
	int mx,sum;
}sb[M];
void ins(int x,int y)
{
	len++;
	sa[len].x=x;
	sa[len].y=y;
	sa[len].next=first[x];
	first[x]=len;
}
int dep[N],son[N],fa[N],size[N],top[N],pos[N];
int root[M],ls[M],rs[M];
int ti=0,num=0;
void dfs1(int u){
    size[u]=1;
    for (int i=first[u];i!=-1;i=sa[i].next){
        int v=sa[i].y;
        if (v!=fa[u]){
            fa[v]=u;
            dep[v]=dep[u]+1;
            dfs1(v);
            size[u]+=size[v];
            if (son[u]==0||size[son[u]]<size[v]){
                son[u]=v;
            }
        }
    }
}

void dfs2(int u,int chain){
    ti++;
    pos[u]=ti; 
    top[u]=chain;
    if (son[u]!=0){
        dfs2(son[u],chain);
    }
    for (int i=first[u];i!=-1;i=sa[i].next){
        int v=sa[i].y;
        if (v!=fa[u]&&v!=son[u]){
            dfs2(v,v);
        }
    }
}
/*=================================预处理================================*/
void update(int x)
{
	sb[x].mx=max(sb[ls[x]].mx,sb[rs[x]].mx);
	sb[x].sum=sb[ls[x]].sum+sb[rs[x]].sum;
}
void change(int &rt,int l,int r,int x,int val)
{
	if(rt==0) rt=++num;
	if(l==r){sb[rt].mx=sb[rt].sum=val;return;}
	int mid=(l+r)/2;
	if(mid>=x) change(ls[rt],l,mid,x,val);
	else change(rs[rt],mid+1,r,x,val);
	update(rt);
}
int get_sum(int &k,int l,int r,int x,int y){
    if (k==0) k=++num;
    if (l==x&&r==y) return sb[k].sum;
    int mid=(l+r)/2;
    if (mid>=y) return get_sum(ls[k],l,mid,x,y);
    else if (mid<x) return get_sum(rs[k],mid+1,r,x,y);
    else return get_sum(ls[k],l,mid,x,mid)+get_sum(rs[k],mid+1,r,mid+1,y);
}
int get_mx(int &rt,int l,int r,int x,int y)
{//printf("!");
	if(rt==0) rt=++num;
	if(l==x&&r==y){return sb[rt].mx;}
	int mid=(l+r)/2;
	if(mid>=y) return get_mx(ls[rt],l,mid,x,y);
	else if(mid<x) return get_mx(rs[rt],mid+1,r,x,y);
	else return max(get_mx(ls[rt],l,mid,x,mid),get_mx(rs[rt],mid+1,r,mid+1,y));
}
/*=========================线段树=================================*/
int solvesum(int cx,int x,int y)
{
	int sum=0;
	//
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		sum+=get_sum(root[cx],1,n,pos[top[x]],pos[x]);
		x=fa[top[x]];
	}//printf("%d %d\n",pos[x],pos[y]);
	if(pos[x]>pos[y]) swap(x,y);
	sum+=get_sum(root[cx],1,n,pos[x],pos[y]);
	return sum;
}
int solvemx(int cx,int x,int y)
{
	int mx=0;//
	while(top[x]!=top[y])
	{
		
		if(dep[top[x]]<dep[top[y]]) swap(x,y);
		mx=max(get_mx(root[cx],1,n,pos[top[x]],pos[x]),mx);
		x=fa[top[x]];
	}
	if(pos[x]>pos[y]) swap(x,y);
	mx=max(get_mx(root[cx],1,n,pos[x],pos[y]),mx);
	return mx;
}
/*=====================树链剖分=================================*/ 
void solve()
{
	char s[9];int x,y;
	for(int i=1;i<=q;i++)
	{
		scanf("%s",s);
		scanf("%d%d",&x,&y);
		if(s[0]=='C')
		{
			if(s[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
			{
				w[x]=y;
				change(root[c[x]],1,n,pos[x],w[x]);
			}
		}
		else
		{
			if(s[1]=='S')//sum
			{
				printf("%d\n",solvesum(c[y],x,y));
			}
			else
			{
				printf("%d\n",solvemx(c[y],x,y));
			}
		}
	}
}
int main()
{
	memset(first,-1,sizeof(first));
	memset(son,0,sizeof(son));
	scanf("%d%d",&n,&q);
	for(int i=1;i<=n;i++)
	scanf("%d%d",&w[i],&c[i]);
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		ins(x,y);ins(y,x);
	}
	dfs1(1);
	dfs2(1,1);
	for(int i=1;i<=n;i++)
	change(root[c[i]],1,n,pos[i],w[i]);
	solve();
}