HYSBZ 1036 树的统计Count（树链剖分）

思路：树链剖分的又一个模板题，不过坑了好久...不知道为什么输入询问的时候用string会RE...

#include<iostream>
#include<cmath>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string>
using namespace std;
const int maxn = 30005;
#define lson i<<1,l,m
#define rson i<<1|1,m+1,r
int n,tot;
int dep[maxn],son[maxn],siz[maxn],fa[maxn],top[maxn],id[maxn];
vector<int>e[maxn];
int val[maxn];
int maxv[maxn*4];
int sum[maxn*4];
int temp[maxn];
void push_up(int i)
{
	maxv[i] = max(maxv[i<<1],maxv[i<<1|1]);
	sum[i] = sum[i<<1]+sum[i<<1|1];
}
void build(int i,int l,int r)
{
	maxv[i]=-0x3f3f3f3f;
	if(l==r)
	{
		maxv[i]=val[l];
		sum[i]=val[l];
		return;
	}
	int m = (l+r)>>1;
	build(lson);
	build(rson);
	push_up(i);
}
void update(int v,int vall,int i,int l,int r)
{

    if(l==r)
	{
		sum[i] = vall;
		maxv[i]=vall;
		return;
	}
	int m = (l+r)>>1;
	if(v <= m)
		update(v,vall,lson);
	else
		update(v,vall,rson);
	push_up(i);
}
int query1(int ql,int qr,int i,int l,int r)
{
	if(ql<=l && qr>=r)
		return sum[i];
	int m = (l+r)>>1;
	int ans = 0;
	if(ql<=m)
		ans+=query1(ql,qr,lson);
	if(qr>m)
		ans+=query1(ql,qr,rson);
	return ans;
}
int query2(int ql,int qr,int i,int l,int r)
{
	if(ql<=l && qr>=r)
		return maxv[i];
	int m = (l+r)>>1;
	int ans = -1e9;
	if(ql<=m)
		ans = max(query2(ql,qr,lson),ans);
	if(qr>m)
		ans=max(query2(ql,qr,rson),ans);
	return ans;

}
void dfs1(int u,int f,int d)
{
	dep[u]=d;
	fa[u]=f;
	son[u]=0;
	siz[u]=1;
	for(int i = 0;i<e[u].size();i++)
	{
		int v = e[u][i];
		if(v==f)
			continue;
		dfs1(v,u,d+1);
		siz[u]+=siz[v];
		if (siz[son[u]] < siz[v])
			son[u]=v;
	}
}
void dfs2(int u,int tp)
{
	top[u]=tp;
	id[u]=++tot;
	if(son[u])
		dfs2(son[u],tp);
	for(int i = 0;i<e[u].size();i++)
	{
		int v = e[u][i];
		if(v==fa[u]|| v==son[u])
			continue;
		dfs2(v,v);
	}
}

int Yougth2(int u,int v)
{
	int tp1 = top[u],tp2 = top[v];
	int ans = -1e9;
	while(tp1!=tp2)
	{
		if(dep[tp1] < dep[tp2])
		{
			swap(tp1,tp2);
			swap(u,v);
		}
		ans = max(query2(id[tp1],id[u],1,1,tot),ans);
		u = fa[tp1];
		tp1 = top[u];
	}
	if(dep[u] > dep[v])
		swap(u,v);
	ans=max(query2(id[u],id[v],1,1,tot),ans);
	return ans;
}

int Yougth1(int u,int v)
{
	int tp1 = top[u],tp2 = top[v];
	int ans = 0;
	while(tp1!=tp2)
	{
		if(dep[tp1] < dep[tp2])
		{
			swap(tp1,tp2);
			swap(u,v);
		}
		ans+=query1(id[tp1],id[u],1,1,tot);
		u = fa[tp1];
		tp1 = top[u];
	}
	if(dep[u] > dep[v])
		swap(u,v);
	ans+=query1(id[u],id[v],1,1,tot);
	return ans;
}
int main()
{
    while(scanf("%d",&n)!=EOF)
	{
		tot = 0;
		memset(val,0,sizeof(val));
		memset(son,0,sizeof(son));
		for(int i = 0;i<=n;i++)
			e[i].clear();
		for(int i =1;i<n;i++)
		{
			int u,v;
			scanf("%d%d",&u,&v);
			e[u].push_back(v);
			e[v].push_back(u);
		}
		for(int i = 1;i<=n;i++)
			scanf("%d",&temp[i]);
		dfs1(1,0,1);
		dfs2(1,1);
		for(int i = 1;i<=n;i++)
			val[id[i]]=temp[i];
		build(1,1,tot);
		int q;
		scanf("%d",&q);
		while(q--)
		{
		    char s[105];
			int a,b;
			scanf("%s%d%d",s,&a,&b);
			if(s[1] == 'M')
			{
                  printf("%d\n",Yougth2(a,b));
			}
			if(s[1]=='S')
			{
                  printf("%d\n",Yougth1(a,b));
			}
			if (s[1]=='H')
			{
                 update(id[a],b,1,1,tot);
			}
		}
	}
}

Description

　　一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。我们将以下面的形式来要求你对这棵树完成
一些操作： I. CHANGE u t : 把结点u的权值改为t II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 注意：从点u到点v的路径上的节点包括u和v本身

Input

　　输入的第一行为一个整数n，表示节点的个数。接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有
一条边相连。接下来n行，每行一个整数，第i行的整数wi表示节点i的权值。接下来1行，为一个整数q，表示操作
的总数。接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。 
对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

Output

　　对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

Sample Input

4
1 2
2 3
4 1
4 2 1 3
12
QMAX 3 4
QMAX 3 3
QMAX 3 2
QMAX 2 3
QSUM 3 4
QSUM 2 1
CHANGE 1 5
QMAX 3 4
CHANGE 3 6
QMAX 3 4
QMAX 2 4
QSUM 3 4

Sample Output

4
1
2
2
10
6
5
6
5
16