BZOJ 1036 树的统计Count 树链剖分

以下内容为转载

树链剖分

　　“在一棵树上进行路径的修改、求极值、求和”乍一看只要线段树就能轻松解决，实际上，仅凭线段树是不能搞定它的。我们需要用到一种貌似高级的复杂算法——树链剖分。
　 树链，就是树上的路径。剖分，就是把路径分类为重链和轻链。
　　记siz[v]表示以v为根的子树的节点数，dep[v]表示v的深度(根深度为1)，top[v]表示v所在的链的顶端节点，fa[v]表示v的父亲，son[v]表示与v在同一重链上的v的儿子节点（姑且称为重儿子），w[v]表示v与其父亲节点的连边（姑且称为v的父边）在线段树中的位置。只要把这些东西求出来，就能用logn的时间完成原问题中的操作。
　　重儿子：siz[u]为v的子节点中siz值最大的，那么u就是v的重儿子。
　　轻儿子：v的其它子节点。
　　重边：点v与其重儿子的连边。
　　轻边：点v与其轻儿子的连边。
　　重链：由重边连成的路径。
　　轻链：轻边。
　　
　　剖分后的树有如下性质：
　　性质1：如果(v,u)为轻边，则siz[u] 2 < siz[v]；
　　*性质2：从根到某一点的路径上轻链、重链的个数都不大于logn。 　　

算法实现：
　　我们可以用两个dfs来求出fa、dep、siz、son、top、w。
　　dfs_1：把fa、dep、siz、son求出来，比较简单，略过。
　　dfs_2：⒈对于v，当son[v]存在（即v不是叶子节点）时，显然有top[son[v]] = top[v]。线段树中，v的重边应当在v的父边的后面，记w[son[v]] = totw+1，totw表示最后加入的一条边在线段树中的位置。此时，为了使一条重链各边在线段树中连续分布，应当进行dfs_2(son[v])；
　　⒉对于v的各个轻儿子u，显然有top[u]=u，并且w[u]=totw+1，进行dfs_2过程。这就求出了top和w。将树中各边的权值在线段树中更新，建链和建线段树的过程就完成了。
修改操作：例如将u到v的路径上每条边的权值都加上某值x。
　　一般人需要先求LCA，然后慢慢修改u、v到公共祖先的边。
　　记f1 = top[u]，f2 = top[v]。
　　当f1 <> f2时：不妨设dep[f1] >= dep[f2]，那么就更新u到f1的父边的权值(logn)，并使u = fa[f1]。
　　当f1 = f2时：u与v在同一条重链上，若u与v不是同一点，就更新u到v路径上的边的权值(logn)，否则修改完成；
　　重复上述过程，直到修改完成。

　　求和、求极值操作：类似修改操作，但是不更新边权，而是对其求和、求极值。
　　

　　如图所示，较粗的为重边，较细的为轻边。节点编号旁边有个红色点的表明该节点是其所在链的顶端节点。边旁的蓝色数字表示该边在线段树中的位置。图中1-4-9-13-14为一条重链。

　　当要修改11到10的路径时。
　　第一次迭代：u = 11，v = 10，f1 = 2，f2 = 10。此时dep[f1] < dep[f2]，因此修改线段树中的5号点，v = 4, f2 = 1；
　　第二次迭代：dep[f1] > dep[f2]，修改线段树中10--11号点。u = 2，f1 = 2；
　　第三次迭代：dep[f1] > dep[f2]，修改线段树中9号点。u = 1，f1 = 1；
　　第四次迭代：f1 = f2且u = v，修改结束。 　　

例题：

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

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
using namespace std;
const int maxm = 30005;
const int INF = 1000005;
#define swap(a,b){int temp=a;a=b;b=temp;}
struct node
{
	int sum, Max, l, r;
}tr[maxm * 4];
vector<int>v[maxm];
int val[maxm], link[maxm], tot, fa[maxm], son[maxm], top[maxm], s[maxm], dep[maxm], n, w[maxm];
void dfs_1(int k, int pre, int ndep);
void dfs_2(int k, int pre, int ntop);
void make(int num, int a, int b);
void update(int num, int id, int xx);
int qmax(int num, int a, int b);
int qsum(int num, int a, int b);
int getmax(int l, int r);
int getsum(int l, int r);
int main()
{
	char str[12];
	int q, i, x, y;
	scanf("%d", &n);
	for (i = 1;i < n;i++)
	{
		scanf("%d%d", &x, &y);
		v[x].push_back(y);
		v[y].push_back(x);
	}
	for (i = 1;i <= n;i++)
		scanf("%d", &val[i]);
	dfs_1(1, 0, 1);
	dfs_2(1, 0, 1);
	make(1, 1, n);
	scanf("%d", &q);
	while (q--)
	{
		scanf("%s%d%d", str, &x, &y);
		if (str[0] == 'C')
			update(1, w[x], y - val[x]), val[x] = y;
		else if (str[1] == 'M')
			printf("%d\n", getmax(x, y));
		else
			printf("%d\n", getsum(x, y));
	}
	return 0;
}
void dfs_1(int k, int pre, int ndep)
{
	dep[k] = ndep, fa[k] = pre;s[k] = 1;
	for (int i = 0;i < v[k].size();i++)
	{
		int xx = v[k][i];
		if (xx == pre)continue;
		dfs_1(xx, k, ndep + 1);
		s[k] += s[xx];
		if (son[k] == 0 || s[xx] > s[son[k]])
			son[k] = xx;
	}
}
void dfs_2(int k, int pre, int ntop)
{
	tot++;
	top[k] = ntop;w[k] = tot;link[tot] = k; 
	if (son[k] == 0) 
		return;
	dfs_2(son[k], k, ntop);
	for (int i = 0;i < v[k].size();i++)
	{
		int xx = v[k][i];
		if (xx == pre || xx == son[k]) continue;
		dfs_2(xx, k, xx);
	}
}
void make(int num, int a, int b)
{
	tr[num].l = a, tr[num].r = b;
	if (a == b)
	{
		tr[num].sum = val[link[a]];
		tr[num].Max = val[link[b]];
		return;
	}
	int mid = (a + b) / 2;
	make(num * 2, a, mid);
	make(num * 2 + 1, mid + 1, b);
	tr[num].sum = tr[num * 2].sum + tr[num * 2 + 1].sum;
	tr[num].Max = max(tr[num * 2].Max, tr[num * 2 + 1].Max);
}
void update(int num, int id, int xx)
{
	if (tr[num].l == tr[num].r)
	{
		tr[num].sum += xx;
		tr[num].Max += xx;
		return;
	}
	int mid = (tr[num].l + tr[num].r) / 2;
	if (mid >= id)	update(num * 2, id, xx);
	else update(num * 2 + 1, id, xx);
	tr[num].sum = tr[num * 2].sum + tr[num * 2 + 1].sum;
	tr[num].Max = max(tr[num * 2].Max, tr[num * 2 + 1].Max);
}
int getsum(int l, int r)
{
	int ans = 0, f1, f2;
	f1 = top[l], f2 = top[r];
	while (f1 != f2)
	{
		if (dep[f1] < dep[f2])
		{
			swap(f1, f2);
			swap(l, r);
		}
		ans += qsum(1, w[f1], w[l]);
		l = fa[f1], f1 = top[l];
	}
	if (dep[l] > dep[r]) ans += qsum(1, w[r], w[l]);
	else ans += qsum(1, w[l], w[r]);
	return ans;
}
int getmax(int l, int r)
{
	int ans = -INF, rev, f1, f2;
	f1 = top[l], f2 = top[r];
	while (f1 != f2)
	{
		if (dep[f1] < dep[f2])
		{
			swap(f1, f2);
			swap(l, r);
		}
		ans = max(ans, qmax(1, w[f1], w[l]));
		l = fa[f1], f1 = top[l];
	}
	if (dep[l] > dep[r]) rev = qmax(1, w[r], w[l]);
	else rev = qmax(1, w[l], w[r]);
	ans = max(ans, rev);
	return ans;
}
int qsum(int num, int a, int b)
{
	int ans = 0;
	if (tr[num].l >= a&&tr[num].r <= b)
		return tr[num].sum;
	int mid = (tr[num].l + tr[num].r) / 2;
	if (a <= mid) ans += qsum(num * 2, a, b);
	if (b > mid) ans += qsum(num * 2 + 1, a, b);
	return ans;
}
int qmax(int num, int a, int b)
{
	int ans = -INF;
	if (tr[num].l >= a&&tr[num].r <= b)
		return tr[num].Max;
	int mid = (tr[num].l + tr[num].r) / 2;
	if (a <= mid) ans = max(ans, qmax(num * 2, a, b));
	if (b > mid) ans = max(ans, qmax(num * 2 + 1, a, b));
	return ans;
}