BZOJ3531 SDOI2014 旅行 - 树链剖分,主席树

题意：给定一棵树，树上每个点有权值和类型。支持：修改某个点的类型；修改某个点的权值；询问某条链上某个类型的点的和/最大值。点数/类型数/询问数<=100000.

分析：

树链剖分，对每个类型的点建立线段树（动态开点）。

note: 忘记写t_query返回值调半天……

莫名其妙地1A

代码：

 #include <bits/stdc++.h>
 using namespace std;

 int a[],s[],dep[],size[],vis[],top[],wson[],fa[][],n,m,t1,t2,t3,t4;
 int sid[],tid[],ind,cnt,ch[][],root[],c[],w[];
 vector <int> g[];

 void dfs1(int p){
     size[p]=;
     vis[p]=;
     for(int i=;i<g[p].size();i++)
         if(vis[g[p][i]]==)  {
             dep[g[p][i]]=dep[p]+;
             fa[g[p][i]][]=p;
             dfs1(g[p][i]);
             size[p]+=size[g[p][i]];
             if(size[g[p][i]]>size[wson[p]]) wson[p]=g[p][i];
         }
 }

 void dfs2(int p){
     vis[p]=;
     sid[p]=++ind;
     tid[ind]=p;
     if(wson[p]) {
         top[wson[p]]=top[p];
         dfs2(wson[p]);
     }
     for(int i=;i<g[p].size();i++)
         if(vis[g[p][i]]==) {
             top[g[p][i]]=g[p][i];
             dfs2(g[p][i]);
         }
 }

 void lca_presolve() {
     for(int i=;i<=;i++)
         for(int j=;j<=n;j++)
             fa[j][i]=fa[fa[j][i-]][i-];
 }

 int lca(int p,int q) {
     if(dep[q]>dep[p]) swap(p,q);
     for(int i=;i>=;i--) if(dep[fa[p][i]]>=dep[q]) p=fa[p][i];
     for(int i=;i>=;i--) if(fa[p][i]-fa[q][i]) p=fa[p][i],q=fa[q][i];
     if(p-q) p=fa[p][],q=fa[q][];
     return max(p,q);
 }

 void init() {
     dep[]=;
     top[]=;
     dfs1();
     memset(vis,,sizeof vis);
     dfs2();
     lca_presolve();
 }

 void modify(int &p,int l,int r,int pos,int k) {
     if(p==) p=++cnt;
     if(l==r) {
         a[p]=k;
         s[p]=k;
     }
     else {
         if(pos<=(l+r)/) modify(ch[p][],l,(l+r)/,pos,k);
         else modify(ch[p][],(l+r)/+,r,pos,k);
         a[p]=max(a[ch[p][]],a[ch[p][]]);
         s[p]=s[ch[p][]]+s[ch[p][]];
     }
 }

 void tmodify(int t,int pos,int k) {
     modify(root[t],,*n,sid[pos],k);
 }

 int queryMax(int p,int l,int r,int ql,int qr) {
     if(p==) return ;
     if(l>qr||r<ql) return ;
     if(l>=ql&&r<=qr) return a[p];
     return max(queryMax(ch[p][],l,(l+r)/,ql,qr),queryMax(ch[p][],(l+r)/+,r,ql,qr));
 }

 int querySum(int p,int l,int r,int ql,int qr) {
     if(p==) return ;
     if(l>qr||r<ql) return ;
     if(l>=ql&&r<=qr) return s[p];
     return querySum(ch[p][],l,(l+r)/,ql,qr)+querySum(ch[p][],(l+r)/+,r,ql,qr);
 }

 int lqueryMax(int t,int anc,int son) {
     int ans=;
     while(dep[top[son]]>dep[anc]) {
         ans=max(ans,queryMax(root[t],,*n,sid[top[son]],sid[son]));
         son=fa[top[son]][];
     }
     ans=max(ans,queryMax(root[t],,*n,sid[anc],sid[son]));
     return ans;
 }

 int lquerySum(int t,int anc,int son) {
     int ans=;
     while(dep[top[son]]>dep[anc]) {
         ans+=querySum(root[t],,*n,sid[top[son]],sid[son]);
         son=fa[top[son]][];
     }
     ans+=querySum(root[t],,*n,sid[anc],sid[son]);
     return ans;
 }

 int tqueryMax(int t,int p,int q) {
     int l=lca(p,q);
     return max(lqueryMax(t,l,p),lqueryMax(t,l,q));
 }

 int tquerySum(int t,int p,int q) {
     int l=lca(p,q);
     return lquerySum(t,l,p)+lquerySum(t,l,q)-w[l]*(c[l]==c[p]);
 }

 string str;

 int main(){
     ios::sync_with_stdio(false);
     cin>>n>>m;
     for(int i=;i<=n;i++) {
         cin>>t1>>t2;
         //tmodify(t2,sid[i],t1);
         c[i]=t2;
         w[i]=t1;
     }
     for(int i=;i<n;i++)  {
         cin>>t1>>t2;
         g[t1].push_back(t2);
         g[t2].push_back(t1);
     }
     init();
     for(int i=;i<=n;i++) {
         tmodify(c[i],i,w[i]);
     }
     for(int i=;i<=m;i++) {
         cin>>str>>t1>>t2;
         if(str=="CC") {
             tmodify(c[t1],t1,);
             tmodify(t2,t1,w[t1]);
             c[t1]=t2;
         }
         if(str=="CW") {
             tmodify(c[t1],t1,);
             tmodify(c[t1],t1,t2);
             w[t1]=t2;
         }
         if(str=="QS") {
             cout<<tquerySum(c[t1],t1,t2)<<endl;
         }
         if(str=="QM") {
             cout<<tqueryMax(c[t1],t1,t2)<<endl;
         }
     }
 }