BZOJ1036 树的统计Count

目录

• BZOJ1036 树的统计Count
• 题解
• ｃｏｄｅ

BZOJ1036 树的统计Count

题目传送门

题解

一道树剖裸题，拿来练练手。。

ｃｏｄｅ

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x%10+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('\n');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
/*================Header Template==============*/
#define PAUSE printf("Press Enter key to continue..."); fgetc(stdin);
const int N=1e5+500;
const int inf=1e7;
#define ls(o) o<<1
#define rs(o) o<<1|1
int n,tot,q;
struct edge {
	int to,nxt;
}E[N<<2];
int head[N],a[N],v[N],idx[N],dep[N],top[N],fa[N],heav[N],sz[N];
/*==================Define Area================*/
namespace SegmentTree {
	struct node {
		int l,r,sz,sum,mx;
	}t[N<<3];
	void Update(int o) {
		t[o].sum=t[ls(o)].sum+t[rs(o)].sum;
		t[o].mx=max(t[ls(o)].mx,t[rs(o)].mx);
	}
	void Build(int o,int l,int r) {
		t[o].l=l;t[o].r=r;t[o].sz=r-l+1;
		if(l==r) {
			t[o].mx=t[o].sum=v[l];
			return ;
		}
		int mid=(l+r)>>1;
		Build(ls(o),l,mid);
		Build(rs(o),mid+1,r);
		Update(o);
		return ;
	}
	void Modify(int o,int pos,int v) {
		if(t[o].l==t[o].r&&t[o].l==pos) {
			t[o].sum=t[o].mx=v;
			return ;
		}
		int mid=(t[o].l+t[o].r)>>1;
		if(mid>=pos) Modify(ls(o),pos,v);
		else Modify(rs(o),pos,v);
		Update(o);
		return ;
	}
	int QueryMx(int o,int l,int r) {
		if(t[o].l>=l&&t[o].r<=r) return t[o].mx;
		int mid=(t[o].l+t[o].r)>>1;
		int ans=-inf;
		if(mid>=l) ans=max(ans,QueryMx(ls(o),l,r));
		if(mid<r) ans=max(ans,QueryMx(rs(o),l,r));
		return ans;
	}
	int QuerySum(int o,int l,int r) {
		if(t[o].l>=l&&t[o].r<=r) return t[o].sum;
		int mid=(t[o].l+t[o].r)>>1;
		int ans=0;
		if(mid>=l) ans+=QuerySum(ls(o),l,r);
		if(mid<r) ans+=QuerySum(rs(o),l,r);
		return ans;
	}
}
using namespace SegmentTree;

namespace poufen {
	void dfs1(int o,int ff,int deep) {
		fa[o]=ff;dep[o]=deep;sz[o]=1;
		int mxson=-1;
		for(int i=head[o];~i;i=E[i].nxt) {
			int to=E[i].to;
			if(to==ff) continue ;
			dfs1(to,o,deep+1);
			sz[o]+=sz[to];
			if(sz[to]>mxson) {
				mxson=sz[to];
				heav[o]=to;
			}
		}
		return ;
	} 
	int cnt=0;
	void dfs2(int o,int topp) {
		idx[o]=++cnt;
		top[o]=topp;
		v[cnt]=a[o];
		if(!heav[o]) return ;
		dfs2(heav[o],topp);
		for(int i=head[o];~i;i=E[i].nxt) {
			int to=E[i].to;
			if(!idx[to]) dfs2(to,to);
		} 
	}
	int TreeSum(int x,int y) {
		int ans=0;
		while(top[x]!=top[y]) {
			if(dep[top[x]]<dep[top[y]]) swap(x,y);
			ans+=QuerySum(1,idx[top[x]],idx[x]);
			x=fa[top[x]];
		}
		if(dep[x]>dep[y]) swap(x,y);
		ans+=QuerySum(1,idx[x],idx[y]);
		return ans;
	}
	int TreeMx(int x,int y) {
		int ans=-inf;
		while(top[x]!=top[y]) {
			if(dep[top[x]]<dep[top[y]]) swap(x,y);
			ans=max(ans,QueryMx(1,idx[top[x]],idx[x]));
			x=fa[top[x]];
		}
		if(dep[x]>dep[y]) swap(x,y);
		ans=max(ans,QueryMx(1,idx[x],idx[y]));
		return ans;
	}
}
using namespace poufen;

void Addedge(int u,int v) {
	E[++tot].to=v;E[tot].nxt=head[u];head[u]=tot;
	E[++tot].to=u;E[tot].nxt=head[v];head[v]=tot;
}

int main() {
	read(n);
	memset(head,-1,sizeof head);
	for(int i=1,u,v;i<n;i++) {
		read(u);read(v);
		Addedge(u,v);
	}
	for(int i=1;i<=n;i++) {
		read(a[i]);
	}
	dfs1(1,1,1);
	dfs2(1,1);
	Build(1,1,n);
	read(q);
	for(int i=1;i<=q;i++) {
		char opt[10];
		scanf("%s",opt);
		if(opt[0]=='C') {
			int pos,v;
			read(pos);read(v);
			Modify(1,idx[pos],v);
		}
		if(opt[0]=='Q'&&opt[1]=='M') {
			int x,y;
			read(x);read(y);
			int ans=TreeMx(x,y);
			printf("%d\n",ans);
		}
		if(opt[0]=='Q'&&opt[1]=='S') {
			int x,y;
			read(x);read(y);
			int ans=TreeSum(x,y);
			printf("%d\n",ans);
		}
	}
	return 0;
}
/*
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
*/