SPOJ 375 Query on a tree（树链剖分）

题意：对于一棵树，每条边有一个权值，有两个操作，修改操作修改一条边的权值，查询操作查询两个节点之间的路径中最大边的权值。

思路：这道题如果是静态的那么lca+倍增就可以解决。

如果动态的修改，那么就可以用树链剖分来解决。

用siz[v]表示以v为根的子树的节点数，dep[v]表示v的深度，top[v]表示v所在的重链的顶端节点，fa[v]表示v的父亲，son[v]表示与v在同一重链上的v的儿子节，w[v]表示v与其父亲节点的连边（姑且称为v的父边）在线段树中的位置。

求出这些东西后，相当于在线段树上进行边的操作，时间复杂度为O(n*logn*logn)。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<stack>
#include<string>
#include<map>
#include<set>
#include<ctime>
#define eps 1e-6
#define LL long long
#define pii (pair<int, int>)
//#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;
const int maxn = 11000;
const int INF = 1000000000;
int maxv[maxn*4];
int query(int o, int L, int R, int ql, int qr) {
	int M = (L+R) >> 1, ans = -1;
	if(ql<=L && qr>=R) return maxv[o];
	if(ql <= M) ans= max(ans, query(o*2, L, M, ql, qr));
	if(M < qr) ans = max(ans, query(o*2+1, M+1, R, ql, qr));
	return ans;
}
void update(int o, int L, int R, int p, int v) {
	int M = (L+R) >> 1;
	if(L == R) maxv[o] = v;
	else {
		if(p <= M) update(o*2, L, M, p, v); else update(o*2+1, M+1, R, p, v);
		maxv[o] = max(maxv[o*2], maxv[o*2+1]);
	} 
}
int n, tot;
vector<int> G[maxn];
int siz[maxn], son[maxn], dep[maxn], top[maxn], fa[maxn], pos[maxn];
void init() {
	for(int i = 1; i <= n; i++) G[i].clear();
	tot = 0;
	memset(maxv, 0, sizeof(maxv));
	memset(son, 0, sizeof(son));
}
void dfs(int cur, int f) {
	siz[cur] = 1;
	int tmp = 0;
	for(int i = 0; i < G[cur].size(); i++) {
		int u = G[cur][i];
		if(u == f) continue;
		dep[u] = dep[cur] + 1;
		fa[u] = cur;
		dfs(u, cur);
		siz[cur] += siz[u];
		if(siz[u] > tmp) son[cur] = u, tmp = siz[u];
	}
}
void dfs2(int cur, int tp) {
	top[cur] = tp;
	pos[cur] = ++tot;
	if(son[cur]) dfs2(son[cur], tp);
	for(int i = 0; i < G[cur].size(); i++) {
		int u = G[cur][i];
		if(u==son[cur] || u==fa[cur]) continue;
		dfs2(u, u);
	}
}
int Find(int u, int v) {
	int ans = 0;
	int fu = top[u], fv = top[v];
	//cout << fu << endl << fv << endl;
	while(fu != fv) {
		if(dep[fu]<dep[fv]) swap(fu, fv), swap(u, v);
		ans = max(ans, query(1, 1, n, pos[fu], pos[u]));
		u = fa[fu]; fu = top[u];
	}
	if(u==v) return ans;
	else if(dep[u]<dep[v]) swap(u, v);
	return max(ans, query(1, 1, n, pos[v]+1, pos[u]));
}

int edge[maxn][3];
int main() {
    //freopen("input.txt", "r", stdin);
	int T; cin >> T;
	while(T--) {
		cin >> n;
		init();
		for(int i = 1; i <= n-1; i++) {
			scanf("%d%d%d", &edge[i][0], &edge[i][1], &edge[i][2]);
			G[edge[i][0]].push_back(edge[i][1]);
			G[edge[i][1]].push_back(edge[i][0]);
		}
		dfs(1, 0);
		dfs2(1, 1);
		//cout << dep[1] << endl << dep[2] << endl;
		//cout << top[1] << endl << top[2] << endl;
		//cout << pos[1] << endl << pos[2] << endl;
		//cout << son[1] << endl << son[2] << endl;
		for(int i = 1; i <= n-1; i++) {
			int& u = edge[i][0];
			int& v = edge[i][1];
			if(dep[u] > dep[v]) swap(u, v);
			update(1, 1, n, pos[v], edge[i][2]);
		}
		//cout << Find(1, 2);
		char op[10];
		while(scanf("%s", op)==1 && op[0]!='D') {
			int u, v; scanf("%d%d", &u, &v);
			if(op[0] == 'Q') {
				printf("%d\n", Find(u, v));
			}
			else {
				update(1, 1, n, pos[edge[u][1]], v);
			}
		}
	}
    return 0;
}