BZOJ1036 [ZJOI2008]树的统计Count 树链剖分

一道很裸的树链剖分题目，将树轻重链剖分后用线段树维护重链，记录和与最大值即可。

对着标程敲……顺便精简了一下。感觉这样就已经很短了。

1A！好久没1A了……虽然是照标程改的不过还是很有成就感的。

//BZOJ1036
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<utility>
using namespace std;
const int MAXN=30010;
const int INF=1e9;
vector<int> graph[MAXN];
int n,in1,in2,w[MAXN],q;
int p[MAXN],size[MAXN],son[MAXN],dep[MAXN],top[MAXN],tid[MAXN],tot;
int l[MAXN<<2],r[MAXN<<2];
pair<int,int> tree[MAXN<<2];
char a[20];
pair<int,int> operator+(const pair<int,int> &s1,const pair<int,int> &s2)
{return make_pair(s1.first+s2.first,max(s1.second,s2.second));}
void DFS(int x)
{
	size[x]=1,son[x]=0;
	int len=graph[x].size();
	for(int i=0;i<len;i++)
	{
		int t=graph[x][i];
		if(t!=p[x])
		{
			p[t]=x,dep[t]=dep[x]+1;
			DFS(t);
			size[x]+=size[t];
			if(size[t]>size[son[x]]) son[x]=t;
		}
	}
}
void build_tree(int x,int anc)
{
	tid[x]=++tot,top[x]=anc,tree[x].second=-INF;
	int len=graph[x].size();
	if(son[x]!=0) build_tree(son[x],anc);
	for(int i=0;i<len;i++)
	{
		int t=graph[x][i];
		if(t!=p[x]&&t!=son[x]) build_tree(t,t);
	}
}
void MakeSegTree(int x,int ll,int rr)
{
	l[x]=ll,r[x]=rr;
	if(ll==rr) return;
	int m=(l[x]+r[x])>>1;
	MakeSegTree(x<<1,ll,m);
	MakeSegTree(x<<1|1,m+1,rr);
}
void SegChange(int x,int pos,int tar)
{
	if(l[x]==r[x]) {tree[x]=make_pair(tar,tar);return;}
	int m=(l[x]+r[x])>>1;
	if(pos<=m) SegChange(x<<1,pos,tar);
	else SegChange(x<<1|1,pos,tar);
	tree[x]=tree[x<<1]+tree[x<<1|1];
}
pair<int,int> SegQuery(int x,int ll,int rr)
{
	if(l[x]==ll&&r[x]==rr) return tree[x];
	int m=(l[x]+r[x])>>1;
	if(rr<=m) return SegQuery(x<<1,ll,rr);
	else if(ll>m) return SegQuery(x<<1|1,ll,rr);
	else return SegQuery(x<<1,ll,m)+SegQuery(x<<1|1,m+1,rr);
}
pair<int,int> TreeQuery(int p1,int p2)
{
	int t1=top[p1],t2=top[p2];
	pair<int,int> con=make_pair(0,-INF);
	while(t1!=t2)
	{
		if(dep[top[p1]]<dep[top[p2]]) {swap(p1,p2);swap(t1,t2);}
		con=con+SegQuery(1,tid[t1],tid[p1]);
		p1=p[t1],t1=top[p1];
	}
	if(dep[p1]<dep[p2]) swap(p1,p2);
	con=con+SegQuery(1,tid[p2],tid[p1]);
	return con;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&in1,&in2);
		graph[in1].push_back(in2);
		graph[in2].push_back(in1);
	}
	for(int i=1;i<=n;i++) scanf("%d",&w[i]);
	DFS(1);
	build_tree(1,1);
	MakeSegTree(1,1,n);
	for(int i=1;i<=n;i++) SegChange(1,tid[i],w[i]);
	scanf("%d",&q);
	for(int cas=1;cas<=q;cas++)
	{
		scanf("%s%d%d",a,&in1,&in2);
		if(a[0]=='C') SegChange(1,tid[in1],in2);
		else if(a[1]=='M') printf("%d\n",TreeQuery(in1,in2).second);
		else printf("%d\n",TreeQuery(in1,in2).first);
	}
	return 0;
}