刷题记录:牛客NC20573[ZJOI2008]树的统计COUNT

传送门:牛客

题目描述:

一棵树上有n个节点，编号分别为1到n，每个节点都有一个权值w。
我们将以下面的形式来要求你对这棵树完成 一些操作： 
I. CHANGE u t : 把结点u的权值改为t 
II. QMAX u v: 询问从点u到点v的路径上的节点的最大权值 I 
II. QSUM u v: 询问从点u到点v的路径上的节点的权值和 
注意：从点u到点v的路径上的节点包括u和v本身
输入:
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
输出;
4
1
2
2
10
6
5
6
5
16

一道经典的树链剖分+线段树的题目

可以使用树链剖分将原本的树形结构分解为线性结构,然后使用线段树进行维护单点修改+区间最大值+区间和即可,只不过代码量较大,需要注意一些细节实现

对于树链剖分+线段树,网上有大量对此进行讲解的博客,此处就不在赘述了

下面是具体的代码部分:

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define root 1,n,1
#define ls rt<<1
#define rs rt<<1|1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
const double eps=1e-8;
#define	int_INF 0x3f3f3f3f
#define ll_INF 0x3f3f3f3f3f3f3f3f
int fa[maxn],Size[maxn],max_son[maxn],dep[maxn];
vector<int>edge[maxn];
void dfs1(int u,int per_u) {
	Size[u]=1;
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v==per_u) continue;
		dep[v]=dep[u]+1;
		dfs1(v,u);
		Size[u]+=Size[v];fa[v]=u;
		if(Size[v]>Size[max_son[u]]) {
			max_son[u]=v;
		}
	}
}
int top[maxn],id[maxn],rev[maxn],tot=0;
void dfs2(int u,int t) {
	top[u]=t;id[u]=++tot;rev[tot]=u;
	if(!max_son[u]) return ;
	dfs2(max_son[u],t);
	for(int i=0;i<edge[u].size();i++) {
		int v=edge[u][i];
		if(v==fa[u]||v==max_son[u]) continue;
		dfs2(v,v);
	}
}
struct Segment_tree{
	int l,r,mx,sum;
}tree[maxn*4];int w[maxn];
void pushup(int rt) {
	tree[rt].mx=max(tree[ls].mx,tree[rs].mx);
	tree[rt].sum=tree[ls].sum+tree[rs].sum;
}
void build(int l,int r,int rt) {
	tree[rt].l=l;tree[rt].r=r;tree[rt].sum=0;tree[rt].mx=-int_INF;
	if(l==r) {
		tree[rt].mx=tree[rt].sum=w[rev[l]];
		return ;
	}
	int mid=(l+r)>>1;
	build(lson);build(rson);
	pushup(rt);
}
void update(int pos,int rt,int val) {
	if(tree[rt].l==pos&&tree[rt].r==pos) {
		tree[rt].sum=tree[rt].mx=val;
		return ;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(pos<=mid) update(pos,ls,val);
	else update(pos,rs,val);
	pushup(rt);
}
int ans_max=-int_INF,ans_sum=0;
void query1(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		ans_max=max(ans_max,tree[rt].mx);
		return ;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) query1(l,r,ls);
	else if(l>mid) query1(l,r,rs);
	else query1(l,mid,ls),query1(mid+1,r,rs);
}
void query2(int l,int r,int rt) {
	if(tree[rt].l==l&&tree[rt].r==r) {
		ans_sum+=tree[rt].sum;
		return ;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) query2(l,r,ls);
	else if(l>mid) query2(l,r,rs);
	else query2(l,mid,ls),query2(mid+1,r,rs);
}
void ask1(int u,int v) {
	while(top[u]!=top[v]) {
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		query1(id[top[u]],id[u],1);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);
	query1(id[u],id[v],1);
}
void ask2(int u,int v) {
 	while(top[u]!=top[v]) {	
		if(dep[top[u]]<dep[top[v]]) swap(u,v);
		query2(id[top[u]],id[u],1);
		u=fa[top[u]];
	}
	if(dep[u]>dep[v]) swap(u,v);	
	query2(id[u],id[v],1);
}
int n;
int main() {
	n=read();
	for(int i=1;i<=n-1;i++) {
		int u=read(),v=read();
		edge[u].push_back(v);
		edge[v].push_back(u);
	}
	for(int i=1;i<=n;i++) w[i]=read();
	dfs1(1,1);dfs2(1,0);
	build(1,tot,1);
	int m=read();char opt[10];
	for(int i=1;i<=m;i++) {
		scanf("%s",opt);
		if(opt[0]=='C') {
			int u=read(),val=read();
			update(id[u],1,val);
		}
		else if(opt[1]=='M'){
			int u=read(),v=read();
			ans_max=-int_INF;
			ask1(u,v);
			printf("%d\n",ans_max);
		}
		else {
			int u=read(),v=read();
			ans_sum=0;
			ask2(u,v);
			printf("%d\n",ans_sum);
		}
	}
	return 0;
}