2022HDU#2 1 树链剖分

题意：

给定一颗单向边构成的树。有$q$次询问，每次给出三个由树上点构成的集合$A,B,C$，问有多少个树上的点$v$符合以下条件：

1.集合$A$至少存在一个点能够到达$v$

2.集合$B$至少存在一个点能够到达$v$

3.$v$能到集合$C$的至少一个点

方法：

由于题目的树给的是多根的，且都能到$1$，我们反存边，使得仅有一个根$1$，按照原来的，每个点都只能向上走到$1$的位置，那么$A$或$B$集合的一个点$u$能出发到达的点应该是$u\to 1$路径上的所有点，那么能到集合$C$的所有点应该是$C$集合下方的点。上面的说法是按照题目的图而讨论的，那么现在按照反存边的方法，$A$或$B$集合出发能到达的点应该是$1\to u$的路径所有点，能到集合$C$的所有点是$u$的子树，树链剖分，然后线段树打上三种标记，同时有三种标记的点就是符合条件的点，统计数量即可。

tips：

一种给线段树打上多个标记并统计的方法：

每一个节点记录管理区间有多少种$i$标记，然后我们暴力递归每种标记都有的区间，找到一个区间（设这个区间管理$[l,r]$），如果三种标记的数量都是$r-l+1$，那么就可以直接返回$r-l+1$，收集所有区间的结果就行。减少时间复杂度的根本就是只递归每种标记都有的区间，存在有一种标记没有的区间一定不会有符合条件的点。

一种多$case$清空线段树的方法：

给上一个清空的懒标记在$1$节点，不要去$build$。（仅使用全部初始化为一个值的清况）

#include<bits/stdc++.h>
using namespace std;

struct way
{
    int to,next;
}edge[200005];
int cnt,head[200005];
void add(int u,int v)
{
    edge[++cnt].to=v;
    edge[cnt].next=head[u];
    head[u]=cnt;
}

//给出三个集合A,B,C，问有多少个点V，能至少到三个集合的至少一个点
vector<int>t(4);
int n,q,son[200005],sum[200005],id[200005],id_;
int top[200005],f[200005];
int tree[800005][4],tag[800005][4],depth[200005],tot[800005];

void dfs1(int u,int fa)
{
    sum[u]=1;
    f[u]=fa;depth[u]=depth[fa]+1;
    for(int i=head[u];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==fa) continue;
        dfs1(v,u);
        sum[u]+=sum[v];
        if(sum[son[u]]<sum[v]) son[u]=v;
    }
}

void dfs(int u,int topf)
{
    top[u]=topf;id[u]=++id_;
    if(son[u]) dfs(son[u],topf);
    for(int i=head[u];i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(v==son[u]||v==f[u]) continue;
        dfs(v,v);
    }
}

inline int ls(int x){return x<<1;}

inline int rs(int x){return x<<1|1;}

inline bool check(int l,int r,int x)
{
    for(int i=1;i<=3;i++)
        if(tree[x][i]!=r-l+1) return false;
    return true;
}

inline void push_up(int x)
{
    for(int i=1;i<=3;i++) tree[x][i]=tree[ls(x)][i]+tree[rs(x)][i];
}

inline void get_tag(int l,int r,int x,int k)
{
    if(k>0)
    {
        tree[x][k]=r-l+1;
        tag[x][k]=true;
    }
    else 
    {
        for(int i=1;i<=3;i++) tree[x][i]=tag[x][i]=false;
        tag[x][k]=true;
    }
}

inline void push_down(int l,int r,int x)
{
    int mid=l+r>>1;
    for(int i=0;i<=3;i++)
    {
        if(!tag[x][i]) continue;
        get_tag(l,mid,ls(x),i);
        get_tag(mid+1,r,rs(x),i);
        tag[x][i]=false;
    }
}

void update(int nl,int nr,int l,int r,int x,int k)
{
    if(nl<=l&&r<=nr)
    {
        if(k>0)
        {
            tree[x][k]=r-l+1;
            tag[x][k]=true;
        }
        else 
        {
            for(int i=1;i<=3;i++) tree[x][i]=tag[x][i]=false;
            tag[x][k]=true;
        }
        return;
    }
    int mid=l+r>>1;push_down(l,r,x);
    if(nl<=mid) update(nl,nr,l,mid,ls(x),k);
    if(nr>mid)  update(nl,nr,mid+1,r,rs(x),k);
    push_up(x);
}

void init()
{
    id_=cnt=0;
    for(int i=1;i<=n;i++) sum[i]=0,son[i]=0,head[i]=0;
}

void update_road(int x,int y,int k)
{
    //树链剖分维护路径
    while(top[x]!=top[y])
    {
        if(depth[top[x]]<depth[top[y]]) swap(x,y);
        update(id[top[x]],id[x],1,n,1,k);
        x=f[top[x]];
    }
    if(id[x]>id[y]) swap(x,y);
    update(id[x],id[y],1,n,1,k);
}

bool cango(int x)
{
    //检查区间是否三种标记都有
    for(int i=1;i<=3;i++)
        if(tree[x][i]==0) return false;
    return true;
}

int getans(int l,int r,int x)
{
    if(check(l,r,x)) return r-l+1;
    if(l==r) return 0;
    int mid=l+r>>1,ret=0; push_down(l,r,x);
    //暴力递归三种标记都有的区间
    if(cango(ls(x))) ret+=getans(l,mid,ls(x));
    if(cango(rs(x))) ret+=getans(mid+1,r,rs(x));
    return ret;
}

void work()
{
    scanf("%d%d",&n,&q);
    init();
    for(int i=2;i<=n;i++)
    {
        int x;scanf("%d",&x);
        add(x,i);
    }
    dfs1(1,0);dfs(1,1);
    while(q--)
    {
        update(1,n,1,n,1,0);
        for(int i=1;i<=3;i++) scanf("%d",&t[i]);
        for(int i=1;i<=2;i++)
        {
            int x;scanf("%d",&x);
            while(t[i]--) update_road(x,1,i);
        }
        while(t[3]--)
        {
            int x;scanf("%d",&x);
            update(id[x],id[x]+sum[x]-1,1,n,1,3);
        }
        printf("%d\n",getans(1,n,1));
    }
}

int main()
{
    int t;scanf("%d",&t);
    while(t--) work();
    return 0;
}