SPOJ - QTREE 树链剖分

题解思路:

时间复杂度：

由于任一轻儿子对应的子树大小要小于父节点所对应子树大小的一半
因此从一个轻儿子沿轻边向上走到父节点后 所对应的子树大小至少变为两倍以上
经过的轻边条数自然是不超过log2Nlog2N的
然后由于重链都是间断的 ((连续的可以合成一条))
所以经过的重链的条数是不超过轻边条数+1+1的
因此经过重链的条数也是loglog级别的

综合可知原命题得证

#include<bits/stdc++.h>
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
using namespace std;
typedef long long ll;
const int mx = 1e4 + 10;
int num[mx],n,head[mx],tot,fa[mx];
int siz[mx],dep[mx],size,top[mx];
int u[mx],maxx[mx<<2],val[mx],cost[mx];
char str[mx/10];
struct node  
{  
    int son,nxt;  
    int num;    
}Edge[mx<<1];  
void AddEdge(int x,int y,int num)  
{  
    Edge[tot].son = y;  
    Edge[tot].nxt = head[x];  
    Edge[tot].num = num;
    head[x] = tot++;  
}  
void dfs1(int x,int f)//求重儿子,深度，树大小等等 
{
	dep[x] = dep[f] + 1;
	siz[x] = 1;
	int p = 0,wei = 0;
	for(int i=head[x];~i;i=Edge[i].nxt)
	{
		int son = Edge[i].son;
		if(son==f) continue;
		fa[son] = x;
		u[Edge[i].num] = son;
		dfs1(son,x);
		siz[x] += siz[son];
		if(siz[son]>siz[wei]) wei = son,p = i;
	}
	//把重儿子换到第一个 
	if(p) swap(Edge[head[x]].num,Edge[p].num),swap(Edge[head[x]].son,Edge[p].son);
}
void dfs2(int x,int f)//重链轻链编号 
{
	num[x] = ++size;
	for(int i=head[x];~i;i=Edge[i].nxt)
	{
		int son = Edge[i].son;
		if(son==f) continue;
		if(i==head[x]) top[son] = top[x];//是不是重儿子 
		else top[son] = son;
		dfs2(son,x);
	}
}
void update(int M,int v,int l,int r,int rt)
{
	if(l==r){
		maxx[rt] = v;
		return ;
	}
	int mid = (l+r)>>1;
	if(M<=mid) update(M,v,lson);
	else update(M,v,rson);
	maxx[rt] = max(maxx[rt<<1],maxx[rt<<1|1]);
}
int query(int L,int R,int l,int r,int rt)
{
	if(L<=l&&R>=r) return maxx[rt];
	int mid = (l+r)>>1;
	int ans = 0;
	if(L<=mid) ans = max(ans,query(L,R,lson));
	if(R>mid) ans = max(ans,query(L,R,rson));
	return ans;
}
int main()
{
	int t,b,a,c;scanf("%d",&t);
	while(t--){
		memset(head,-1,sizeof(head));
		memset(maxx,0,sizeof(maxx));
		size = tot = 0;
		top[1] = 1;
		scanf("%d",&n);
		for(int i=1;i<n;i++){
			scanf("%d%d%d",&a,&b,&c);
			AddEdge(a,b,i);
			AddEdge(b,a,i);
			val[i] = c;
		}
		dfs1(1,0),dfs2(1,0);
		for(int i=1;i<n;i++) update(num[u[i]],val[i],1,n,1);
		int x,y;
		while(scanf("%s",str)){
			if(str[0]=='D') break;
			scanf("%d%d",&x,&y);
			if(str[0]=='Q'){
				int f1 = top[x],f2 = top[y],ans = 0;
				while(f1!=f2){//log(n)刨链更新 
					if(dep[f1]>dep[f2]){
						 ans = max(ans,query(num[f1],num[x],1,n,1));
						 x = fa[f1],f1 = top[x];
					}else{
						ans = max(ans,query(num[f2],num[y],1,n,1));
						y = fa[f2],f2 = top[y];
					}
				}
				if(dep[x]!=dep[y]){
					if(dep[x]<dep[y]) ans = max(ans,query(num[x]+1,num[y],1,n,1));
					else ans = max(ans,query(num[y]+1,num[x],1,n,1));
				}
				printf("%d\n",ans);
			}else{
				update(num[u[x]],y,1,n,1);
			}
		}
	}
	return 0;
}