BZOJ-1036 [ZJOI2008]树的统计

树链剖分模版题。

#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <cctype>
#include <cmath>
#define rep(i, l, r) for(int i=l; i<=r; i++)
#define clr(x, c) memset(x, c, sizeof(x))
#define travel(x) for(edge *p=fir[x]; p; p=p->n)
#define k(x) Key[x]
#define t(x) Tree[x]
#define s(x) Size[x]
#define b(x) Belong[x]
#define low(x) Lower[x]
#define dep(x) Deep[x]
#define h(x) Head[x]
#define l(x) Left[x]
#define r(x) Right[x]
#define w(x) Where[x]
#define maxn 30009
#define maxp 5000009
#define inf 0x7fffffff
using namespace std;
inline int read()
{
	int x=0, f=1; char ch=getchar();
	while (!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}
	while (isdigit(ch)) x=x*10+ch-'0', ch=getchar();
	return x*f;
}
struct edge{int y; edge *n;} e[maxn*2], *fir[maxn], *pt=e;
inline void AddE(int x, int y)
{
	pt->y=y, pt->n=fir[x], fir[x]=pt++;
	pt->y=x, pt->n=fir[y], fir[y]=pt++;
}
int Tree[maxn], Size[maxn], Belong[maxn], Lower[maxn], Deep[maxn], Head[maxn], Where[maxn], cnt, Key[maxn];
int Left[maxp], Right[maxp], sum[maxp], big[maxp], z, L, R;
int n, d[maxn], h[maxn], now;


void dfs(int x)
{
	int maxs=-inf, maxl=0;
	travel(x) if (p->y!=h[x])
	{
		h[p->y]=x, d[p->y]=d[x]+1; dfs(p->y);
		if (s(b(p->y))>maxs) maxs=s(b(p->y)), maxl=p->y;
	}
	if (maxl)
		b(x)=b(maxl), w(x)=w(maxl)+1, s(b(x))++, dep(b(x))--;
	else
		cnt++, b(x)=cnt, w(x)=1, s(cnt)=1, low(cnt)=x, dep(cnt)=d[x];
	travel(x) if (p->y!=h[x] && p->y!=maxl) h(b(p->y))=x;
}
void BuildT(int&k, int l, int r, int t)
{
	if (l==r){sum[t]=big[t]=k(k), k=h[k]; return;}
	int mid=(l+r)>>1;
	BuildT(k, l, mid, l(t)=++z), BuildT(k, mid+1, r, r(t)=++z);
	sum[t]=sum[l(t)]+sum[r(t)], big[t]=max(big[l(t)], big[r(t)]);
}
inline void Build()
{
	dfs(1); h(b(1))=0;
	rep(i, 1, cnt) now=low(i), BuildT(now, 1, s(i), t(i)=++z);
}


void Edit(int k, int y, int l, int r, int t)
{
	if (l==r){sum[t]=big[t]=y; return;}
	int mid=(l+r)>>1;
	if (k<=mid) Edit(k, y, l, mid, l(t)); else Edit(k, y, mid+1, r, r(t));
	sum[t]=sum[l(t)]+sum[r(t)], big[t]=max(big[l(t)], big[r(t)]);
}
inline void Change(int x, int y){k(x)=y; Edit(w(x), y, 1, s(b(x)), t(b(x)));}

void QueryM(int l, int r, int t)
{
	if (L<=l && r<=R) {now=max(now, big[t]); return;}
	int mid=(l+r)>>1;
	if (L<=mid) QueryM(l, mid, l(t));
	if (mid<R) QueryM(mid+1, r, r(t));
}
inline void Qmax(int x, int y)
{
	now=-inf; while (b(x)!=b(y))
	{
		if (dep(b(x))<dep(b(y))) swap(x, y);
		L=w(x), R=s(b(x)), QueryM(1, s(b(x)), t(b(x))), x=h(b(x));
	}
	if (d[x]<d[y]) swap(x, y);
	L=w(x), R=w(y), QueryM(1, s(b(x)), t(b(x)));
	printf("%d
", now);
}
void QueryS(int l, int r, int t)
{
	if (L<=l && r<=R) {now+=sum[t]; return;}
	int mid=(l+r)>>1;
	if (L<=mid) QueryS(l, mid, l(t));
	if (mid<R) QueryS(mid+1, r, r(t));
}
inline void Qsum(int x, int y)
{
	now=0; while (b(x)!=b(y))
	{
		if (dep(b(x))<dep(b(y))) swap(x, y);
		L=w(x), R=s(b(x)), QueryS(1, s(b(x)), t(b(x))), x=h(b(x));
	}
	if (d[x]<d[y]) swap(x, y);
	L=w(x), R=w(y), QueryS(1, s(b(x)), t(b(x))); 
	printf("%d
", now);
}


int main()
{
	n=read();
	rep(i, 1, n-1) {int x=read(), y=read(); AddE(x, y);}
	rep(i, 1, n) k(i)=read();
	Build();
	int q=read(); char ch[10];
	rep(i, 1, q)
	{
		scanf("%s", ch); int x=read(), y=read();
		if (ch[1]=='H') Change(x, y);
		if (ch[1]=='S') Qsum(x, y);
		if (ch[1]=='M') Qmax(x, y);
	}
}