树链剖分：洛谷 P2590 [ZJOI2008]树的统计

一共三部分：树剖，线段树，LCA

难打主要是因为代码长

#include<stdio.h>
#include<string.h>
#include<ctype.h>
#define mid (l+r>>1)
#define N 30005
#define M 19
#define INF 1128481604
int n,q;
int mx[N<<2],sum[N<<2];
int head[N],next[N<<1],to[N<<1];
int fa[N][M+1],son[N],dep[N],size[N],top[N],num[N],pos[N];
int swap(int &a,int &b) {
	static int t;
	t=a, a=b, b=t;
}
int max(int a,int b) {
	return a>b ? a : b;
}
void add(int a,int b) {
	static int cnt=0;
	to[++cnt]=b;
	next[cnt]=head[a];
	head[a]=cnt;
}
void dfs1(int t,int f,int d) {
	fa[t][0]=f, dep[t]=d, size[t]=1;
	for(int i=head[t];i;i=next[i]) {
		if(to[i]!=fa[t][0]) {
			dfs1(to[i],t,d+1);
			size[t]+=size[to[i]];
			if(size[to[i]]>size[son[t]]) son[t]=to[i];
		}
	}
}
void dfs2(int t,int tp) {
	static int cnt=0;
	top[t]=tp, num[t]=++cnt, pos[cnt]=t;
	if(son[t]) dfs2(son[t],tp);
	for(int i=head[t];i;i=next[i]) {
		if(to[i]!=fa[t][0]&&to[i]!=son[t]) dfs2(to[i],to[i]);
	}
}
void setpre() {
	for(int i=1;i<=M;i++)
		for(int j=1;j<=n;j++)
			fa[j][i]=fa[fa[j][i-1]][i-1];
}
int lca(int a,int b) {
	if(dep[a]>dep[b]) swap(a,b);
	int d=dep[b]-dep[a];
	for(int i=M;i>=0;i--) {
		if(d&(1<<i)) b=fa[b][i];
	}
	if(a==b) return a;
	for(int i=M;i>=0;i--) {
		if(fa[a][i]!=fa[b][i]) a=fa[a][i], b=fa[b][i];
	}
	return fa[a][0];
}
void pushup(int t) {
	mx[t]=max(mx[t<<1],mx[t<<1|1]);
	sum[t]=sum[t<<1]+sum[t<<1|1];
}
void update(int t,int k,int v,int l,int r) {
	if(l==r) mx[t]=v, sum[t]=v;
	else {
		if(k<=mid) update(t<<1,k,v,l,mid);
		else update(t<<1|1,k,v,mid+1,r);
		pushup(t);
	}
}
int querysum(int t,int L,int R,int l,int r) {
	if(L<=l&&r<=R) return sum[t];
	int v=0;
	if(L<=mid) v+=querysum(t<<1,L,R,l,mid);
	if(R>mid) v+=querysum(t<<1|1,L,R,mid+1,r);
	return v;
}
int querymx(int t,int L,int R,int l,int r) {
	if(L<=l&&r<=R) return mx[t];
	int v=-INF;
	if(L<=mid) v=max(v,querymx(t<<1,L,R,l,mid));
	if(R>mid) v=max(v,querymx(t<<1|1,L,R,mid+1,r));
	return v;
}
int getsum(int t,int g,int b) {
	int v=0;
	while(top[t]!=top[g]) {
		v+=querysum(1,num[top[t]],num[t],1,n);
		t=fa[top[t]][0];
	}
	if(t!=g||!b) v+=querysum(1,num[g]+b,num[t],1,n);
	return v;
}
int getmx(int t,int g) {
	int v=-INF;
	while(top[t]!=top[g]) {
		v=max(v,querymx(1,num[top[t]],num[t],1,n));
		t=fa[top[t]][0];
	}
	v=max(v,querymx(1,num[g],num[t],1,n));
	return v;
}

int main() {
	memset(mx,188,sizeof mx);
	int a,b,g;
	char c;
	scanf("%d",&n);
	for(int i=1;i<n;i++) {
		scanf("%d%d",&a,&b);
		add(a,b), add(b,a);
	}
	dfs1(1,0,1);
	dfs2(1,1);
	setpre();
	for(int i=1;i<=n;i++) {
		scanf("%d",&a);
		update(1,num[i],a,1,n);
	}
	scanf("%d",&q);
	while(q--) {
		while(isspace(getchar()));
		c=getchar();
		while(isalpha(getchar()));
		scanf("%d%d",&a,&b);
		if(c=='H') update(1,num[a],b,1,n);
		else {
			g=lca(a,b);
			if(c=='S') printf("%d\n",getsum(a,g,0)+getsum(b,g,1));
			else printf("%d\n",max(getmx(a,g),getmx(b,g)));
		}
	}
}

看别人的代码发现把LCA和询问写在一起可以快1倍

以上