HDU4685 Prince and Princess强连通

Posted 2020-10-10 codeg

题意：

有n个王子和m个公主，每个王子都会喜欢若干个公主，也就是王子只跟自己喜欢的公主结婚，公主就比较悲惨， 跟谁结婚都行。然后输出王子可能的结婚对象，必须保证王子与任意这些对象中的一个结婚，都不会影响到剩余的王子的配对数，也就是不能让剩余的王子中突然有一个人没婚可结了。

思路：

对每个没有匹配的公主，建一个虚拟的王子，让他们变成匹配，然后由这个虚拟王子向每个公主建边。对每个没有匹配的王子，建一个虚拟的公主，让他们变成匹配，然后每个王子向这个虚拟公主建边。求一个Tarjan,判断是否为1个强连通分量即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#define maxn 2005

using namespace std;

int dfn[maxn],low[maxn],vi[maxn],Stack[maxn],head[maxn],id[maxn],n,e,lab,num,top,ans[maxn];
int graphic[510][510],m,match[maxn],cho[maxn],mm;

struct Edge{
    int u,v,next;
}edge[400005];

vector<int> q[maxn],girl[maxn];

void init()
{
    memset(dfn,0,sizeof(dfn));
    memset(low,0,sizeof(low));
    memset(vi,0,sizeof(vi));
    memset(head,-1,sizeof(head));
    memset(id,0,sizeof(id));
    memset(graphic,0,sizeof(graphic));
    memset(cho,0,sizeof(cho));
    memset(match,-1,sizeof(match));
    memset(ans,0,sizeof(ans));
    for(int i=0;i<maxn;i++)
        q[i].clear(),girl[i].clear();
    e=lab=num=top=mm=0;
}

void add(int u,int v)
{
    edge[e].u=u,edge[e].v=v,edge[e].next=head[u],head[u]=e++;
}

int dfs(int u)//匈牙利算法
{
    int i;
    for(i=1;i<=m;i++)
    {
        if(!vi[i]&&graphic[u][i])
        {
            vi[i]=1;
            if(match[i]==-1||dfs(match[i]))
            {
                match[i]=u;
                return 1;
            }
        }
    }
    return 0;
}

int Tarjan(int u)
{
    dfn[u]=low[u]=++lab;
    vi[u]=1;
    Stack[top++]=u;
    int i,v;
    for(i=head[u];i!=-1;i=edge[i].next)
    {
        v=edge[i].v;
        if(!dfn[v])
        {
            Tarjan(v);
            low[u]=min(low[u],low[v]);
        }
        if(vi[v])
            low[u]=min(low[u],dfn[v]);
    }
    if(low[u]==dfn[u])
    {
        ++num;
        do{
            i=Stack[--top];
            vi[i]=0;
            id[i]=num;
        }while(i!=u);
    }
    return 0;
}

void solve()
{
    int i,j;
    for(i=1;i<=n+m+mm;i++)
        if(!dfn[i])
        Tarjan(i);
    for(i=n+1;i<=n+m;i++)
    {
        q[id[i]].push_back(i-n);
    }
    int cnt=0;
    for(i=1;i<=n;i++)
    {
        int size=q[id[i]].size();
        cnt=0;
        for(j=0;j<size;j++)
        {
            if(graphic[i][q[id[i]][j]])
                ans[cnt++]=q[id[i]][j];
        }
        printf("%d",cnt);
        for(j=0;j<cnt;j++)
            printf(" %d",ans[j]);
        printf("\n");
    }
}

int main()
{
    int a,b,i,j,t,Case=0;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d%d",&n,&m);
        for(i=1;i<=n;i++)
        {
        scanf("%d",&a);
        while(a--)
        {
            scanf("%d",&b);
            add(i,b+n);
            graphic[i][b]=1;
        }
        }
        int num=0;
        memset(match,-1,sizeof(match));
        for(i=1;i<=n;i++)
        {
            memset(vi,0,sizeof(vi));
            if(dfs(i))
                num++;
        }
        mm=0;
        for(i=1;i<=m;i++)
        {
            if(match[i]==-1)
            {
                mm++;
                add(n+m+mm,i+n);
                add(i+n,n+m+mm);
                for(j=1;j<=m;j++)
                    add(n+m+mm,j+n);
            }
            else
            {
                cho[match[i]]=1;
            }
            add(i+n,match[i]);
        }
        for(i=1;i<=n;i++)
        {
            if(cho[i]==0)
            {
                mm++;
                add(i,n+m+mm);
                add(n+m+mm,i);
                for(j=1;j<=n;j++)
                    add(j,n+m+mm);
            }
        }
        memset(vi,0,sizeof(vi));
        printf("Case #%d:\n",++Case);
        solve();
    }
    return 0;
}

View Code