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

题目描述

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。

我们将以下面的形式来要求你对这棵树完成一些操作：

I. CHANGE u t : 把结点u的权值改为t

II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值

III. QSUM u v: 询问从点u到点v的路径上的节点的权值和

注意：从点u到点v的路径上的节点包括u和v本身

输入输出格式

输入格式：

 

输入文件的第一行为一个整数n，表示节点的个数。

接下来n – 1行，每行2个整数a和b，表示节点a和节点b之间有一条边相连。

接下来一行n个整数，第i个整数wi表示节点i的权值。

接下来1行，为一个整数q，表示操作的总数。

接下来q行，每行一个操作，以“CHANGE u t”或者“QMAX u v”或者“QSUM u v”的形式给出。

 

输出格式：

 

对于每个“QMAX”或者“QSUM”的操作，每行输出一个整数表示要求输出的结果。

 

输入输出样例

输入样例#1： 复制

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

输出样例#1： 复制

4
1
2
2
10
6
5
6
5
16

说明

对于100％的数据，保证1<=n<=30000，0<=q<=200000；中途操作中保证每个节点的权值w在-30000到30000之间。

 

记录第一次完整的写下来树链剖分，虽然只是个模版题 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn=60000+5;
const int inf=2e9;
char op[10];
int head[maxn],w[maxn],siz[maxn],top[maxn],son[maxn],dep[maxn],fa[maxn],tid[maxn],rnk[maxn],tot,cnt,n,q;
struct edge{int to,next;}e[maxn];
inline void addedge(int u,int v)
{
	cnt++;e[cnt].next=head[u];head[u]=cnt;e[cnt].to=v;
}
struct segment
{
	int sum[maxn<<2],maxx[maxn<<2];
	inline void build(int o,int l,int r)
	{
		if(l==r){sum[o]=maxx[o]=w[rnk[l]];return ;}
		int m=(l+r)>>1;
		build(o<<1,l,m);build(o<<1|1,m+1,r);
		sum[o]=sum[o<<1]+sum[o<<1|1];
		maxx[o]=max(maxx[o<<1],maxx[o<<1|1]);
	}
	inline int query1(int o,int l,int r,int ql,int qr)// max
	{
		if(l>qr||r<ql) return -inf;
		if(ql<=l&&r<=qr) return maxx[o];
		int m=(l+r)>>1;
		return max(query1(o<<1,l,m,ql,qr),query1(o<<1|1,m+1,r,ql,qr));
	}
	inline int query2(int o,int l,int r,int ql,int qr)// sum
	{
		if(l>qr||r<ql) return 0;
		if(ql<=l&&r<=qr)return sum[o];
		int m=(l+r)>>1;
		return query2(o<<1,l,m,ql,qr)+query2(o<<1|1,m+1,r,ql,qr);
	}
	inline void update(int o,int l,int r,int x,int t)
	{
		if(l==r){maxx[o]=sum[o]=t;return ;}
		int m=(l+r)>>1;
		if(x<=m)update(o<<1,l,m,x,t);
		else update(o<<1|1,m+1,r,x,t);
		sum[o]=sum[o<<1]+sum[o<<1|1];
		maxx[o]=max(maxx[o<<1],maxx[o<<1|1]);
	}
}st;
inline void dfs1(int u,int fat)
{
	son[u]=-1;siz[u]=1;
	for(int i=head[u];i;i=e[i].next)
	{
		if(!dep[e[i].to])
		{
			dep[e[i].to]=dep[u]+1;
			fa[e[i].to]=u;dfs1(e[i].to,u);
			siz[u]+=siz[e[i].to];
			if(son[u]==-1||siz[e[i].to]>siz[son[u]])son[u]=e[i].to;
		}
	}
}
inline void dfs2(int u,int t)
{
	top[u]=t;tot++;
	tid[u]=tot;rnk[tot]=u;
	if(son[u]==-1)return;
	dfs2(son[u],t);
	for(int i=head[u];i;i=e[i].next)
		if(e[i].to!=son[u]&&e[i].to!=fa[u])dfs2(e[i].to,e[i].to);
}
inline int querymax(int x,int y)
{
	int res=-inf,fx=top[x],fy=top[y];
	while(fx!=fy)
	{
		if(dep[fx]>=dep[fy]) res=max(res,st.query1(1,1,n,tid[fx],tid[x])),x=fa[fx];
		else res=max(res,st.query1(1,1,n,tid[fy],tid[y])),y=fa[fy];
		fx=top[x];fy=top[y];
	}
	if(x!=y)
	{
		if(tid[x]<tid[y])res=max(res,st.query1(1,1,n,tid[x],tid[y]));
		else res=max(res,st.query1(1,1,n,tid[y],tid[x]));
	} else res=max(res,st.query1(1,1,n,tid[x],tid[y]));
	return res;
}
inline int querysum(int x,int y)
{
	int res=0,fx=top[x],fy=top[y];
	while(fx!=fy)
	{
		if(dep[fx]>=dep[fy])res+=st.query2(1,1,n,tid[fx],tid[x]),x=fa[fx];
		else res+=st.query2(1,1,n,tid[fy],tid[y]),y=fa[fy];
		fx=top[x];fy=top[y];
	}
	if(x!=y)
	{
		if(tid[x]<tid[y]) res+=st.query2(1,1,n,tid[x],tid[y]);
		else res+=st.query2(1,1,n,tid[y],tid[x]);
	} else res+=st.query2(1,1,n,tid[x],tid[y]);
	return res;
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<n;i++)
	{
		int u,v;scanf("%d%d",&u,&v); addedge(u,v);addedge(v,u);
	}
	for(int i=1;i<=n;i++)scanf("%d",&w[i]);
	dep[1]=1;dfs1(1,-1);dfs2(1,1);st.build(1,1,n);
    scanf("%d",&q);
    while(q--)
    {
    	int u,v;scanf("%s%d%d",op,&u,&v);
    	if(op[0]=='C')st.update(1,1,n,tid[u],v);
        if(op[0]=='Q'&&op[1]=='M')printf("%d\n",querymax(u,v));
        if(op[0]=='Q'&&op[1]=='S')printf("%d\n",querysum(u,v));
    }
	return 0;
}