树链剖分：洛谷 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
,next[N<<1],to[N<<1];
int fa
[M+1],son
,dep
,size
,top
,num
,pos
;
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倍
以上