2019河南省CCPC省赛：咕咕的搜索序列(树的dfs序)

题目链接：http://acm.zzuli.edu.cn/problem.php?id=2525
题目大意：

思路：
1：出现 i 后，不会出现 i 的子节点。
2：

所以：对于任意两个次序：u， v。

如果v已经染色，BAD GUGU。
找到u, v的LCA。把v及其子树全部染色。把LCA到u的那棵子树全部染色。

注意：初始化，不然会T

#include <bits/stdc++.h>
#define LL long long
using namespace std;
 
int cut, N=1000005, b[1000005], v[1000005];
int fa[1000005], d[1000005], head[1000005];//d深度， fa父节点， b次序，v颜色
 
struct E
{
    int v;
    int next;
 
}e[2000005];
 
void init()
{
    fa[1]=1;
    cut=-1;
    for(int i=0;i<=N;i++)
    {
        head[i]=-1;
        v[i]=0;
    }
}
 
void addEdge(int u, int v)
{
    cut++;
    e[cut].next=head[u];
    e[cut].v=v;
    head[u]=cut;
}
 
int dfs(int u, int tot)
{
    d[u]=tot;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int to=e[i].v;
        if(to!=fa[u])
        {
            fa[to]=u;
            dfs(to, tot+1);
        }
    }
}
 
int LCA(int u, int v)//返回LCA到u的第一个子树节点
{
    while(d[fa[u]]>d[fa[v]])
    {
        u=fa[u];
    }
    while(d[fa[v]]>d[fa[u]])
    {
        v=fa[v];
    }
    while(fa[v]!=fa[u])
    {
        u=fa[u]; v=fa[v];
    }
    return u;
}
 
int RS(int u)//染色
{
    if(v[u])
    {
        return 0;
    }
    v[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next)
    {
        int to=e[i].v;
        if(to!=fa[u]&&!v[to])
        {
            RS(to);
        }
    }
}
 
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        int n, m, k=0;
        scanf("%d%d",&n,&m);
        N=n;
        for(int i=2;i<=n;i++)
        {
            int x;
            scanf("%d",&x);
            addEdge(i, x);
            addEdge(x, i);
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d",&b[i]);
        }
        dfs(1, 0);
        for(int i=0;i<m;i++)
        {
            if(v[b[i]])
            {
                k=1;
                break;
            }
            RS(b[i]);//染v
            if(i!=0)
            {
                int L=LCA(b[i-1], b[i]);
                RS(L);//染LCA->u的子树
            }
        }
        if(k)
        {
            printf("BAD GUGU\n");
        }
        else
        {
            printf("NOT BAD\n");
        }
    }
}