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('
');}
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
",ans);
}
if(opt[0]=='Q'&&opt[1]=='S') {
int x,y;
read(x);read(y);
int ans=TreeSum(x,y);
printf("%d
",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
*/