[HZOI 2015]树黑白

【题目描述】

给定一棵树，要求维护以下操作：

1、M u 将u节点反色

2、Q u 查询u到所有黑色节点距离和

【输入格式】

第一行n,m 表示节点总数和操作次数

之后n-1行，每行u，v表示两个端点，w表示边权

之后m行，操作如题意

n，m<=200000,边权<=1000

一开始所有点均为白色

【输出格式】

输出每次Q的答案

【样例输入】

7 5

2 1 144

3 2 361

4 3 236

5 3 697

6 2 140

7 5 718

Q 4

M 4

Q 5

Q 1

Q 7

【样例输出】

0

933

741

1651

题解：

和上一题差不多，多了一个修改操作。

还是动态点分，每个节点储存两个数，分别表示整棵子树到根和到根节点的父节点的距离和。

容斥一下就好。

 //Never forget why you start
 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<cmath>
 #include<algorithm>
 #define inf (2147483647)
 using namespace std;
 int n,m,a[],lim;
 struct node{
   int next,to,dis;
 }edge[];
 int head[],size;
 void putin(int from,int to,int dis){
   size++;
   edge[size].next=head[from];
   edge[size].to=to;
   edge[size].dis=dis;
   head[from]=size;
 }
 int fa[][],dis[],depth[];
 void dfs1(int r,int father){
   int i;
   fa[r][]=father;
   depth[r]=depth[father]+;
   for(i=head[r];i!=-;i=edge[i].next){
     int y=edge[i].to;
     if(y!=father){
       dis[y]=dis[r]+edge[i].dis;
       dfs1(y,r);
     }
   }
 }
 void make(){
   lim=log(n)/log();
   for(int i=;i<=lim;i++)
     for(int j=;j<=n;j++)
       fa[j][i]=fa[fa[j][i-]][i-];
 }
 int LCA(int x,int y){
   if(depth[x]<depth[y])swap(x,y);
   for(int i=lim;i>=;i--)
     if(depth[fa[x][i]]>=depth[y])
       x=fa[x][i];
   if(x!=y){
     for(int i=lim;i>=;i--)
       if(fa[x][i]!=fa[y][i])
     x=fa[x][i],y=fa[y][i];
     x=fa[x][];
     y=fa[y][];
   }
   return x;
 }
 int dist(int x,int y){
   int lca=LCA(x,y);
   return dis[x]+dis[y]-dis[lca]*;
 }
 int tot,root,vis[],d[],ff[],cnt[];
 void getroot(int r,int father){
   int i;
   cnt[r]=;d[r]=;
   for(i=head[r];i!=-;i=edge[i].next){
     int y=edge[i].to;
     if(y!=father&&!vis[y]){
       getroot(y,r);
       cnt[r]+=cnt[y];
       d[r]=max(d[r],cnt[y]);
     }
   }
   d[r]=max(d[r],tot-cnt[r]);
   if(d[root]>d[r])root=r;
 }
 void buildtree(int r,int father){
   int i,all=tot;
   ff[r]=father;vis[r]=;
   for(i=head[r];i!=-;i=edge[i].next){
     int y=edge[i].to;
     if(!vis[y]){
       if(cnt[y]>cnt[r])cnt[y]=all-cnt[r];tot=cnt[y];
       root=;getroot(y,r);buildtree(root,r);
     }
   }
 }
 int p[][];
 void insert(int x,int v){
   int i;
   for(i=x;ff[i];i=ff[i]){
     int len=dist(x,ff[i]);
     p[i][]+=len*v;
     p[ff[i]][]+=len*v;
   }
   for(i=x;i;i=ff[i])cnt[i]+=v;
 }
 int find(int x){
   int i,ans=p[x][];
   for(i=x;ff[i];i=ff[i]){
     int len=dist(x,ff[i]);
     ans+=p[ff[i]][];
     ans-=p[i][];
     ans+=(cnt[ff[i]]-cnt[i])*len;
   }
   return ans;
 }
 void clean(){
   memset(head,-,sizeof(head));
   size=;
 }
 int main(){
   freopen("A_Tree.in","r",stdin);
   freopen("A_Tree.out","w",stdout);
   int i,j;
   clean();
   scanf("%d%d",&n,&m);
   for(i=;i<n;i++){
     int u,v,l;
     scanf("%d%d%d",&u,&v,&l);
     putin(u,v,l);
     putin(v,u,l);
   }
   dfs1(,);make();
   tot=n;root=;d[]=inf;
   getroot(,);buildtree(root,);
   for(i=;i<=n;i++)if(!ff[i]){root=i;break;}
   char s[];
   memset(cnt,,sizeof(cnt));
   while(m--){
     scanf("%s",s);
     if(s[]=='Q'){
       int x;
       scanf("%d",&x);
       printf("%d\n",find(x));
     }
     else{
       int x;
       scanf("%d",&x);
       a[x]^=;
       if(a[x])insert(x,);
       else insert(x,-);
     }
   }
   return ;
 }