【HDU 5221】 Occupation 【树链剖分+线段树】

题意:

    给你一棵树,每个节点有个点权

    有三种操作:

         1.把u到v的路径全部标记为1

         2.把x标记为0

         3.把x的子树标记为1

    让你输出每次操作后的标记为1的点权和

题解：

     点权和可以用线段树(线段树维护树链剖分之后重新分配id之后的权值)维护,操作1用树链剖分对线段树一段一段的更新,操作2直接线段树更新,操作3可以在树链剖分之后直接update(tid[x],tid[x]+siz[x]-1),因为树链剖分后,x和其子树的下标是连续的

     线段树flag=1代表这一段全部被标记为1了,要往下push,因为成段更新只有成段更新为1的情况，清空的话单点更新是

#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define PB push_back
#define MP make_pair
#define ll long long
#define MS(a,b) memset(a,b,sizeof(a))
#define LL (rt<<1)
#define RR (rt<<1|1)
#define lson l,mid,LL
#define rson mid+1,r,RR
#define pii pair<int,int>
#define pll pair<ll,ll>
#define lb(x) (x&(-x))
void In(){freopen("in.in","r",stdin);}
void Out(){freopen("out.out","w",stdout);}
const int N=1e5+10;
const int M=3e5+10;
const int Mbit=1e6+10;
const int inf=0x3f3f3f3f;
const ll mod=1e9+7;
int id,NE;
int dep[N],head[N],son[N],siz[N],father[N],top[N],tid[N],mark[N];
struct Edge
{
    int v,next;
}edge[N<<1];
void init(int n)
{
    NE=0;
    for(int i=1;i<=n;i++)head[i]=-1;
}
void add(int u,int v)
{
    edge[NE].v=v;
    edge[NE].next=head[u];
    head[u]=NE++;
}
void dfs1(int u,int fa)
{
    dep[u]=dep[fa]+1;
    son[u]=0;
    father[u]=fa;
    siz[u]=1;
    for(int i=head[u];~i;i=edge[i].next){
        int v=edge[i].v;
        if(v==fa)continue;
        dfs1(v,u);
        siz[u]+=siz[v];
        if(siz[son[u]]<siz[v])son[u]=v;
    }
}
void dfs2(int u,int fa)
{
    tid[u]=id++;
    mark[tid[u]]=u;
    top[u]=fa;
    if(son[u]!=0)dfs2(son[u],fa);
    for(int i=head[u];~i;i=edge[i].next){
        int v=edge[i].v;
        if(v==father[u]||v==son[u])continue;
        dfs2(v,v);
    }
}
int val[N],n,sum[N];
int flag[N<<2],tree[N<<2];
void up(int rt)
{
    tree[rt]=tree[LL]+tree[RR];
}
void down(int rt,int l,int r)
{
    int mid=r+l>>1;
    if(flag[rt]){
        flag[LL]=flag[RR]=1;
        tree[LL]=sum[mid]-sum[l-1];
        tree[RR]=sum[r]-sum[mid];
        flag[rt]=0;
    }
}
void build(int l,int r,int rt)
{
    tree[rt]=flag[rt]=0;
    if(l==r)return;
    int mid=l+r>>1;
    build(lson);
    build(rson);
    up(rt);
}
void update(int l,int r,int rt,int L,int R,int x)
{
    if(L<=l&&r<=R){
        flag[rt]=x;
        tree[rt]=x*(sum[r]-sum[l-1]);
        return;
    }
    down(rt,l,r);
    int mid=l+r>>1;
    if(L<=mid) update(lson,L,R,x);
    if(R>mid)  update(rson,L,R,x);
    up(rt);
}
void color(int u,int v)
{
    while(top[u]!=top[v]){
        if(dep[top[u]]>dep[top[v]])swap(u,v);
        update(1,n,1,tid[top[v]],tid[v],1);
        v=father[top[v]];
    }
    if(tid[u]>tid[v])swap(u,v);
    update(1,n,1,tid[u],tid[v],1);
}
int main()
{
    int T,kase=0,m,a,b,op;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        init(n);
        for(int i=1;i<=n;i++)scanf("%d",&val[i]);
        for(int i=1;i<n;i++){
            int u,v;scanf("%d%d",&u,&v);
            add(u,v);add(v,u);
        }
        siz[0]=dep[1]=0;id=1;
        dfs1(1,1);
        dfs2(1,1);
        build(1,n,1);
        for(int i=1;i<=n;i++)
            sum[i]=sum[i-1]+val[mark[i]];
        scanf("%d",&m);
        for(int i=1;i<=m;i++){
            scanf("%d",&op);
            if(op==1){
                scanf("%d%d",&a,&b);
                color(a,b);
            }
            else if(op==2){
                scanf("%d",&a);
                update(1,n,1,tid[a],tid[a],0);
            }
            else if(op==3){
                scanf("%d",&a);
                update(1,n,1,tid[a],tid[a]+siz[a]-1,1);
            }
            printf("%d\n",tree[1]);
        }
    }
    return 0;
}