BZOJ3589动态树

**错误改了一上午。

先做熟练泼粪

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

By：大奕哥

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