BZOJ - 1036 树的统计Count (树链剖分+线段树)

题目链接

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const int N=3e4+,inf=0x3f3f3f3f;
 int hd[N],ne,n,k,fa[N],son[N],siz[N],dep[N],top[N],dfn[N],rnk[N],tot,a[N],sum[N<<],mx[N<<];
 struct E {int v,nxt;} e[N<<];
 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}
 void dfs1(int u,int f,int d) {
     fa[u]=f,son[u]=,siz[u]=,dep[u]=d;
     for(int i=hd[u]; ~i; i=e[i].nxt) {
         int v=e[i].v;
         if(v==fa[u])continue;
         dfs1(v,u,d+),siz[u]+=siz[v];
         if(siz[v]>siz[son[u]])son[u]=v;
     }
 }
 void dfs2(int u,int tp) {
     top[u]=tp,dfn[u]=++tot,rnk[dfn[u]]=u;
     if(!son[u])return;
     dfs2(son[u],top[u]);
     for(int i=hd[u]; ~i; i=e[i].nxt) {
         int v=e[i].v;
         if(v==fa[u]||v==son[u])continue;
         dfs2(v,v);
     }
 }
 #define ls u<<1
 #define rs u<<1|1
 void pu(int u) {sum[u]=sum[ls]+sum[rs],mx[u]=max(mx[ls],mx[rs]);}
 void build(int u=,int l=,int r=tot) {
     if(l==r) {sum[u]=mx[u]=a[l]; return;}
     int mid=(l+r)>>;
     build(ls,l,mid),build(rs,mid+,r);
     pu(u);
 }
 void upd(int p,int x,int u=,int l=,int r=tot) {
     if(l==r) {sum[u]=mx[u]=x; return;}
     int mid=(l+r)>>;
     p<=mid?upd(p,x,ls,l,mid):upd(p,x,rs,mid+,r);
     pu(u);
 }
 void qry(int L,int R,int& _sum,int& _mx,int u=,int l=,int r=tot) {
     if(l>=L&&r<=R) {_sum+=sum[u],_mx=max(_mx,mx[u]); return;}
     if(l>R||r<L)return;
     int mid=(l+r)>>;
     qry(L,R,_sum,_mx,ls,l,mid),qry(L,R,_sum,_mx,rs,mid+,r);
 }
 void change(int u,int x) {upd(dfn[u],x);}
 void ask(int u,int v,int& _sum,int& _mx) {
     _sum=,_mx=~inf;
     for(; top[u]!=top[v]; u=fa[top[u]]) {
         if(dep[top[u]]<dep[top[v]])swap(u,v);
         qry(dfn[top[u]],dfn[u],_sum,_mx);
     }
     if(dep[u]<dep[v])swap(u,v);
     qry(dfn[v],dfn[u],_sum,_mx);
 }
 int main() {
     memset(hd,-,sizeof hd),ne=;
     scanf("%d",&n);
     for(int i=; i<n; ++i) {
         int u,v;
         scanf("%d%d",&u,&v);
         addedge(u,v);
         addedge(v,u);
     }
     tot=,dfs1(,,),dfs2(,);
     for(int i=; i<=n; ++i)scanf("%d",&a[dfn[i]]);
     build();
     char s[];
     scanf("%d",&k);
     while(k--) {
         int a,b;
         scanf("%s%d%d",s,&a,&b);
         if(s[]=='H')change(a,b);
         else {
             int _sum,_mx;
             ask(a,b,_sum,_mx);
             if(s[]=='S')printf("%d\n",_sum);
             else printf("%d\n",_mx);
         }
     }
     return ;
 }