hdu4370 Werewolf 并查集+dfs

题意：狼人可能说假话 村民一定说真话 求金水和铁狼的个数

 

因为狼人可能说假话 所以没有谁一定是村民

枚举两个人之间的关系可知当a->b是狼 b->a是人时 b一定是狼

由此 如果一个环上只有一条狼人边 其他都是村民边 则被指的狼一定是狼 并且环外还有人指狼是村民则也是狼

 

将村民边用并查集维护 将狼人边单独储存 然后每次判断狼人边的两端是否在一棵树内 如果在则可以确定

 

注意 狼人边储存的时候用的是反向边 就是a->b我们存成b->a 这样在搜索的过程中才能保证不遗漏所有的指向狼的人边

#include <bits/stdc++.h>
///唯一能确定是狼的情况是环上有人指是狼 并且其他边都是人
using namespace std;
const int N=1e5+100;
int father[N],sum[N],n;
int Find(int x)
{
    return father[x]=x==father[x]?x:Find(father[x]);
}
void unite(int x,int y)
{
    int tx=Find(x);
    int ty=Find(y);
    if(tx==ty)return;
    father[tx]=ty;
}
vector<int>w[N];
vector<int>v[N];
int vis[N];
void init()
{
    for(int i=1;i<=n;i++)
    {
        vis[i]=0;
        father[i]=i;
        sum[i]=1;
        w[i].clear();
        v[i].clear();
    }
}
int ans;
void dfs1(int x)///找狼边
{
    int l=w[x].size();
    for(int i=0;i<l;i++)
    {
        int u=w[x][i];
        if(Find(u)==Find(x))
            ans+=sum[u];///这个点作为人边的出点的次数
    }
}
int dfs(int x)///更新每个点作为人边的出点的次数
{
    vis[x]=1;
    int l=v[x].size();
    for(int i=0;i<l;i++)
    {
        int u=v[x][i];
        if(vis[u])
        sum[x]+=sum[u];
        else
        sum[x]+=dfs(u);
    }
    return sum[x];
}
int main()
{
    ios::sync_with_stdio(false);
    int T,x;
    string s;
    cin>>T;
    while(T--)
    {
        cin>>n;
        init();
        for(int i=1;i<=n;i++)
        {
            cin>>x>>s;
            if(s[0]=='w')
            {
                w[i].push_back(x);
            }
            else
            {
                unite(x,i);
                v[x].push_back(i);
            }
        }
        ans=0;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i])
            {
                dfs(i);
            }
        }
        for(int i=1;i<=n;i++)
            dfs1(i);
        cout<<0<<" "<<ans<<endl;
    }
    return 0;
}
/*
10
8
2 w
3 vv
4 vv
5 vv
1 vv
2 vv
2 vv
2 vv
*/