POJ - 1904 King's Quest （强连通分量+完美匹配）

链接：https://cn.vjudge.net/problem/POJ-1904

题意：有n个王子和n个女孩，每个王子都有若干个喜欢的女孩，王子需要娶喜欢的女孩，一个王子只能娶一个女孩。现在给出一个初始的完美匹配，也就是n个王子娶n个女孩的一种方案。输出对于每个王子，他可以娶的女孩的编号的集合。王子娶这个集合的任一个女孩，其它王子都能娶到自己喜欢的女孩。

思路：题目给出了一个完美匹配，我们建图。将所有的王子指向他们喜欢的女孩，然后将给出的完美匹配的女孩指向匹配的王子。然后，求强连通分量。王子可以娶的女孩的集合就是在同一强连通分量的女孩。为什么？假设王子a的初始完美匹配的是女孩x，王子b的初始完美匹配的是女孩y,那么如果a和y在同一个强连通分量里，那么b和x肯定也在该强联通分量里，因为有边a->x，x->a，b->y，y->b。初始完美匹配中的王子和女孩肯定在同一个强连通分量（他们本身就是一个连通分量），也就是说一个强连通分量中的王子和女孩的个数是相同的。既然他们之间可以两两互达，那么王子娶同一强连通分量中的哪个女孩都不会有影响。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 4e3+10;
const int M =1e6+10;
struct node
{
    int to,nxt;
}g[M];
int head[N],cnt;
int dfn[N],low[N],id,sta[N],top,color[N],cl;
bool vis[N];
int n;
int x,num,ans[N];
inline int Read()
{
    int res=0,ch,flag=0;
    if((ch=getchar())=='-')
        flag=1;
    else if(ch>='0'&&ch<='9')
        res=ch-'0';
    while((ch=getchar())>='0'&&ch<='9')
        res=res*10+ch-'0';
    return flag?-res:res;
}
inline void Write(int a)
{
    if(a>9)
        Write(a/10);
    putchar(a%10+'0');
}
void Init()
{
    cnt=top=id=cl=0;
    for(int i=1;i<=2*n;i++)
        head[i]=-1,dfn[i]=vis[i]=0;
}
void add(int u,int v)
{
    g[cnt].to=v; g[cnt].nxt=head[u]; head[u]=cnt++;
}
void tarjan(int u)
{
    int v;
    dfn[u]=low[u]=++id; sta[++top]=u; vis[u]=1;
    for(int i=head[u];i!=-1;i=g[i].nxt)
    {
        v=g[i].to;
        if(!dfn[v])
        {
            tarjan(v);
            if(low[u]>low[v]) low[u]=low[v];
        }
        else if(vis[v]&&low[u]>dfn[v])
            low[u]=dfn[v];
    }
    if(dfn[u]==low[u])
    {
        color[u]=++cl;
        while(sta[top]!=u)
        {
            color[sta[top]]=cl; vis[sta[top--]]=0;
        }
        vis[sta[top--]]=0;
    }
}
int main()
{
    int u,v;
    while(~scanf("%d",&n))
    {
        Init();
        for(u=1;u<=n;u++)
        {
            x=Read();//scanf("%d",&x);
            while(x--)
            {
                v=Read();//scanf("%d",&v);
                add(u,v+n);
            }
        }
        for(u=1;u<=n;u++)
        {
            v=Read();//scanf("%d",&v);
            add(v+n,u);
        }
        for(int i=1;i<=2*n;i++)
            if(!dfn[i]) tarjan(i);
        for(u=1;u<=n;u++)
        {
            num=0;
            for(int i=head[u];i!=-1;i=g[i].nxt)
            {
                v=g[i].to;
                if(color[u]==color[v])
                    ans[++num]=v-n;
            }
            sort(ans+1,ans+num+1);
            Write(num);//printf("%d",num);
            for(int i=1;i<=num;i++)
            {
                putchar(' ');
                Write(ans[i]);
                //printf(" %d",ans[i]);
            }
            putchar('\n');//printf("\n");
        }
    }
    return 0;
}