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

题解：

树剖+线段树

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

主要是打代码。。。。。

#include
#include
#include
#include
#include
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]]=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=y) return get_mx(ls[rt],l,mid,x,y);
	else if(midpos[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]]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