[Sdoi2014]旅行 题解

题目大意：

　　给出一个n个点的树，和m次操作。每个点有颜色和权值。 每次操作分4种 1：修改一个点的颜色 2：修改一个点的权值 3：询问从x到y的路径上，和x相同颜色的点的权值和（保证x，y同颜色） 4：询问从x到y的路径上，和x相同颜色的点的权值最大值（保证x，y同颜色）

思路：

　　树链剖分，用线段树来维护和以及最大值。

代码：

 #include<cstdio>
 #include<iostream>
 #define M 3000009
 using namespace std;

 int cnt,dfn,num,n,m,w[M],c[M],to[M<<],next[M<<],head[M],deep[M],size[M],pa[M],id[M],top[M],root[M],sum[M<<],mx[M<<],lc[M<<],rc[M<<];

 void ins(int x,int y)
 {
      to[++cnt]=y,next[cnt]=head[x],head[x]=cnt;
 }

 void dfs1(int x)
 {
      size[x]=;
      for (int i=head[x];i;i=next[i])
          if (pa[x]!=to[i])
          {
               deep[to[i]]=deep[x]+,pa[to[i]]=x;
               dfs1(to[i]),size[x]+=size[to[i]];
          }
 }

 void dfs2(int x,int chain)
 {
      int i,k=;
      id[x]=++dfn,top[x]=chain;
      for (i=head[x];i;i=next[i])
          if (deep[to[i]]>deep[x] && size[to[i]]>size[k]) k=to[i];
      if (!k) return; dfs2(k,chain);
      for (i=head[x];i;i=next[i])
          if (pa[x]!=to[i] && k!=to[i]) dfs2(to[i],to[i]);
 }

 void up_date(int k)
 {
      sum[k]=sum[lc[k]]+sum[rc[k]];
      mx[k]=max(mx[lc[k]],mx[rc[k]]);
 }

 void change(int &cur,int l,int r,int x,int val)
 {
      if (!cur) cur=++num;
      if (l==r) { sum[cur]=mx[cur]=val; return; }
      int mid=l+r>>;
      if (x<=mid) change(lc[cur],l,mid,x,val);
      else change(rc[cur],mid+,r,x,val);
      up_date(cur);
 }

 int ask_sum(int cur,int L,int R,int l,int r)
 {
     if (!cur) return ;
     if (L==l && R==r) return sum[cur];
     int mid=L+R>>;
     if (r<=mid) return ask_sum(lc[cur],L,mid,l,r);
     if (l>mid) return ask_sum(rc[cur],mid+,R,l,r);
     return ask_sum(lc[cur],L,mid,l,mid)+ask_sum(rc[cur],mid+,R,mid+,r);
 }

 int ask_max(int cur,int L,int R,int l,int r)
 {
     if (!cur) return ;
     if (L==l && R==r) return mx[cur];
     int mid=L+R>>;
     if (r<=mid) return ask_max(lc[cur],L,mid,l,r);
     if (l>mid) return ask_max(rc[cur],mid+,R,l,r);
     return max(ask_max(lc[cur],L,mid,l,mid),ask_max(rc[cur],mid+,R,mid+,r));
 }

 int Ask_Sum(int x,int y,int z)
 {
     int ans=;
     for (;top[x]!=top[y];x=pa[top[x]])
     {
         if (deep[top[x]]<deep[top[y]]) swap(x,y);
         ans+=ask_sum(root[z],,n,id[top[x]],id[x]);
     }
     if (deep[x]>deep[y]) swap(x,y);
     return ans+ask_sum(root[z],,n,id[x],id[y]);
 }

 int Ask_Max(int x,int y,int z)
 {
     int ans=-;
     for (;top[x]!=top[y];x=pa[top[x]])
     {
         if (deep[top[x]]<deep[top[y]]) swap(x,y);
         ans=max(ans,ask_max(root[z],,n,id[top[x]],id[x]));
     }
     if (deep[x]>deep[y]) swap(x,y);
     return max(ans,ask_max(root[z],,n,id[x],id[y]));
 }

 int main()
 {
     scanf("%d%d",&n,&m);
     int i,x,y;
     for (i=;i<=n;i++) scanf("%d%d",&w[i],&c[i]);
     for (i=;i<n;i++) scanf("%d%d",&x,&y),ins(x,y),ins(y,x);
     dfs1(),dfs2(,);
     for (i=;i<=n;i++) change(root[c[i]],,n,id[i],w[i]);
     for (i=;i<=m;i++)
     {
         char ch[];
         scanf("%s%d%d",ch,&x,&y);
         if (ch[]=='C')
             if (ch[]=='C') change(root[c[x]],,n,id[x],),change(root[y],,n,id[x],w[x]),c[x]=y;
             else change(root[c[x]],,n,id[x],y),w[x]=y;
         else if (ch[]=='S') printf("%d\n",Ask_Sum(x,y,c[x]));
              else printf("%d\n",Ask_Max(x,y,c[x]));
     }scanf("%d",&n);
     return ;
 }