题目链接:http://acm.zzuli.edu.cn/problem.php?id=2525
题目大意:
思路:
1:出现 i 后,不会出现 i 的子节点。
2:
所以:对于任意两个次序:u, v。
如果v已经染***AD 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");
}
}
}