BZOJ3589动态树

**错误改了一上午。

先做熟练泼粪

k<=5，因此我们可以模拟这个过程，在线段树上把标记建出来然后pushup时候更新就好了。

By：大奕哥

 #include<bits/stdc++.h>
 using namespace std;
 const int N=;
 struct tree{
     int l,lz,ll;long long s,ret;
 }t[N<<];
 int head[N],cnt,n,id,bel[N],pos[N],size[N],d[N],f[N],son[N],ed[N];
 struct node{
     int to,nex;
 }e[N<<];
 void add(int x,int y)
 {
     e[++cnt].to=y;e[cnt].nex=head[x];head[x]=cnt;
 }
 void dfs(int x)
 {
     size[x]=;
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(y==f[x])continue;
         d[y]=d[x]+;f[y]=x;
         dfs(y);
         if(size[y]>size[son[x]])son[x]=y;
         size[x]+=size[y];
     }
     return;
 }
 void dfs2(int x,int chain)
 {
     pos[x]=++id;bel[x]=chain;
     if(son[x])dfs2(son[x],chain);
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         if(y==f[x]||y==son[x])continue;
         dfs2(y,y);
     }
     ed[x]=id;
     return;
 }
 void update(int x,int z)
 {
     t[x].lz+=z;
     t[x].s+=z*t[x].l;
 }
 void update2(int x,int z)
 {
     t[x].ll=z;
     t[x].ret=t[x].s*z;
 }
 void pushdown(int x)
 {
     if(t[x].lz)
     {
         update(x<<,t[x].lz);
         update(x<<|,t[x].lz);
         t[x].lz=;
     }
     if(t[x].ll!=-)
     {
         update2(x<<,t[x].ll);
         update2(x<<|,t[x].ll);
         t[x].ll=-;
     }
     return;
 }
 void pushup(int x)
 {
     t[x].s=t[x<<].s+t[x<<|].s;
     t[x].ret=t[x<<].ret+t[x<<|].ret;
 }
 void build(int x,int l,int r)
 {
     t[x].ll=-;
     if(l==r){
         t[x].l=;return;
     }
     int mid=l+r>>;
     t[x].l=r-l+;
     build(x<<,l,mid);build(x<<|,mid+,r);
 }
 void paint(int x,int l,int r,int L,int R,int w)
 {
     if(t[x].ll==w)return;
     if(l==L&&r==R){
         update2(x,w);
         return;
     }
     pushdown(x);
     int mid=l+r>>;
     if(mid<L)paint(x<<|,mid+,r,L,R,w);
     else if(mid>=R)paint(x<<,l,mid,L,R,w);
     else paint(x<<,l,mid,L,mid,w),paint(x<<|,mid+,r,mid+,R,w);
     pushup(x);
     return;
 }
 void change(int x,int l,int r,int L,int R,int c)
 {
     if(l==L&&r==R)
     {
         update(x,c);
         return;
     }
     pushdown(x);
     int mid=l+r>>;
     if(mid<L)change(x<<|,mid+,r,L,R,c);
     else if(mid>=R)change(x<<,l,mid,L,R,c);
     else change(x<<,l,mid,L,mid,c),change(x<<|,mid+,r,mid+,R,c);
     pushup(x);
 }
 void color(int x,int y,int c)
 {
     while(bel[x]!=bel[y])
     {
         if(d[bel[x]]<d[bel[y]])swap(x,y);
         paint(,,n,pos[bel[x]],pos[x],c);
         x=f[bel[x]];
     }
     if(d[x]<d[y])swap(x,y);
     paint(,,n,pos[y],pos[x],c);
     return;
 }
 int main()
 {
     int x,y,q,ff,k;
     scanf("%d",&n);
     for(int i=;i<n;++i)
     {
         scanf("%d%d",&x,&y);
         add(x,y);add(y,x);
     }
     dfs();dfs2(,);
     build(,,n);
     scanf("%d",&q);
     for(int i=;i<=q;++i)
     {
         scanf("%d",&ff);
         if(ff==)
         {
             scanf("%d%d",&x,&y);
             change(,,n,pos[x],ed[x],y);
         }
         else{
             scanf("%d",&k);
             for(int j=;j<=k;++j)
             {
                 scanf("%d%d",&x,&y);
                 color(x,y,);
             }
             printf("%d\n",t[].ret&((1ll<<)-));
             update2(,);
         }
     }
     return ;
 }