2018年全国多校算法寒假训练营练习比赛（第四场） E.通知小弟(Tarjan,缩点)

题目描述

在战争时期，A国派出了许多间谍到其他国家去收集情报。因为间谍需要隐秘自己的身份，所以他们之间只是单向联系。所以，某个间谍只能单向联系到一部分的间谍。同时，间谍也不知道跟他联系的是谁。
HA是间谍们的老大，但他也只能联系到部分的间谍。HA现在有一项命令有告诉所有的间谍。HA想要知道他至少要告诉多少个他能联系上的间谍才能通知到所有的间谍。

输入描述:

有多个测试数据。 对于每个测试数据：
第一行为一个整数n，m（0＜n,m<=500）代表间谍的数量和HA能通知到的间谍的数量（间谍的编号为1-n）；
第二行为m个用空格隔开的整数xi,代表HA能通知到的间谍的编号；
第三行到第n+2行，每一行第一个整数ai(0<=ai＜n)表示第i-2个间谍能单向联系到的间谍数。之后有ai个用空格隔开的整数，表示间谍i-2能单向联系到的间谍的编号。

输出描述:

输出一行，此行中有一个整数，代表HA至少需要联系的间谍数。如果HA不能通知到所有间谍，输出-1。

输入

3 2
1 2
1 2
1 1
0

输出

-1

输入

3 1
1
2 2 3
0
0

输出

1

思路

题意很明确,给出了能直接通知到的人，然后给出了谁能通知到谁，证明这是一个有向图.

我们可以先利用Tarjan给这个图缩点，缩成若干的强连通分量，在每一个强连通分量离，可以通知到当前分量的任意一个人就证明这个强连通分量的每个人都可以知道答案。

那么，缩点之后，标记出哪一个强连通分量可以直接被通知到。我们再找出缩点后的图中入度为0的点，如果当前的点可以被直接通知到，那么就计数，如果当前的点入度为0还不能被通知到，那么就直接输出-1.

代码

#include <cstdio>
#include <cstring>
#include <cctype>
#include <stdlib.h>
#include <string>
#include <map>
#include <iostream>
#include <set>
#include <stack>
#include <cmath>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
#define mem(a,b) memset(a,b,sizeof(a))
typedef long long ll;
const int N=500+2;
const int M=N*N;
int dfn[N],low[N],vis[N],times,cnt;
int color[N];//缩点用
int in[N];//入度的个数
stack<int>s;//标记一个强连通分量中的点
int first[N],tot;
struct edge
{
    int v,next;
} e[M];
void add_edge(int u,int v)
{
    e[tot].v=v;
    e[tot].next=first[u];
    first[u]=tot++;
}
void init()
{
    mem(dfn,0);
    mem(low,0);
    mem(vis,0);
    mem(first,-1);
    mem(color,0);
    mem(in,0);
    times=0;
    tot=0;
    cnt=0;
    while(!s.empty())s.pop();
}
void tarjan(int u)
{
    vis[u]=1;
    s.push(u);
    dfn[u]=low[u]=++times;
    for(int i=first[u]; ~i; i=e[i].next)
    {
        int v=e[i].v;
        if(!dfn[v])
        {
            tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        else if(vis[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(low[u]==dfn[u])
    {
        cnt++;
        while(1)
        {
            int now=s.top();
            s.pop();
            color[now]=cnt;
            vis[now]=0;
            if(now==u) break;
        }
    }
}
int teach[N];//能通知到的人
int book[N];//标记可不可以通知到当前的联通分量
int main()
{
    int n,m,x,v;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        mem(teach,0);
        mem(book,0);
        for(int i=1; i<=m; i++)
        {
            scanf("%d",&x);
            teach[x]=1;
        }
        for(int u=1; u<=n; u++)
        {
            scanf("%d",&x);
            while(x--)
            {
                scanf("%d",&v);
                add_edge(u,v);
            }
        }
        for(int i=1; i<=n; i++)
            if(!dfn[i])
                tarjan(i);
        for(int u=1; u<=n; u++)
        {
            for(int j=first[u]; ~j; j=e[j].next)
            {
                v=e[j].v;
                if(color[u]!=color[v])
                    in[color[v]]++;//v的入度
            }
        }
        for(int i=1; i<=n; i++)
        {
            x=color[i];
            if(teach[i])//能直接通知i
                book[x]=1;
        }
        int flag=1,ans=0;
        for(int i=1; i<=cnt; i++)
        {
            if(!in[i])//入度为0的点
            {
                if(book[i])
                    ans++;
                else
                {
                    flag=0;
                    break;
                }
            }
        }
        if(!flag) puts("-1");
        else printf("%d\n",ans);
    }
    return 0;
}