BJTUOJ 1850 预言家困境（dfs）

Description

狼人杀是一款桌面游戏，主题是好人和狼人两个阵营的对抗。

现在有$n$个人聚在一起玩狼人杀。所有人指认一个人是狼人，好人说的一定说真话，狼人一定说谎。

在好人阵营中，有一位预言家，可以分辨一部分人是好人还是狼人。$Alice$抽到了预言家，除了知道她自己是预言家外，又额外知道了$p+q$个人的身份。现在$Alice$希望通过已有信息，尽可能多的确认每个人的身份。

Input

第一行有一个整数$n$，代表玩家总数$(2\le n\le 5\cdot 10^5)$

第二行有$n−1$个整数$c_i$，代表除$Alice$外，其他人指认的人的编号。编号从$1$开始，$Alice$的编号为$n(1\le c_i\le n)$

第三行的第一个整数$p$ ，代表$Alice$知道了 $p$个人的身份是好人，其余$p$个整数分别代表他们的编号$(p\ge 0)$

第四行的第一个整数$q$，代表$Alice$知道了$q$个人的身份是狼人，其余$q$个整数分别代表他们的编号$(q\ge 0，p+q<n)$

数据保证没有冲突。

Output

输出包含两行。

第一行的第一个整数$p′$，代表$Alice$可以确认的其他好人数量为$p′$，其余$p′$ 个整数分别代表他们的编号，按编号从小到大输出。

第二行的第一个整数$q′$，代表$Alice$可以确认的狼人数量为$q′$，其余$q′$个整数分别代表他们的编号，按编号从小到大输出。

Sample Input

4
2 3 2
1 1
1 2

Sample Output

2 1 3
1 2

Solution

第$i$个人说第$c_i$个人是狼人即说明$i$和$c_i$必然不是同一阵营，以此建图，从已经确认的点开始黑白染色即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=500005;
int n,p,q,vis[maxn],temp;
vector<int>a,b,c[maxn];
void dfs(int x,int sta)
{
    if(vis[x])return ;
    vis[x]=sta;
    for(int i=0;i<c[x].size();i++)
    {
        int y=c[x][i];
        if(sta==1)dfs(y,2);
        else dfs(y,1);
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)c[i].clear();
        for(int i=1;i<n;i++)
        {
            scanf("%d",&temp);
            c[i].push_back(temp),c[temp].push_back(i);
        }
        dfs(n,1);
        scanf("%d",&p);
        while(p--)
        {
            scanf("%d",&temp);
            dfs(temp,1);
        }
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d",&temp);
            dfs(temp,2);
        }
        for(int i=1;i<n;i++)
            if(vis[i]==1)a.push_back(i);
            else if(vis[i]==2)b.push_back(i);
        printf("%d",a.size());
        for(int i=0;i<a.size();i++)printf(" %d",a[i]);
        printf("\n%d",b.size());
        for(int i=0;i<b.size();i++)printf(" %d",b[i]);
        printf("\n");
    }
    return 0;
}