链接:https://cn.vjudge.net/problem/HDU-4685
题意:T组样例。有n个王子和n个公主,每个王子都有若干个喜欢的公主,王子需要娶喜欢的公主,一个王子只能娶一个公主。输出对于每个王子,他可以娶的公主的编号的集合。王子娶这个集合的任一个公主,其他王子与公主的最大匹配数不变。
思路:这道题和poj1904https://blog.csdn.net/birdmanqin/article/details/99673878很是像,只不过那道题给出了一个完美匹配,这个题没有,那么我们就构造一个完美匹配,解法就和那个题一样了。首先先跑一次匈牙利,求出一个最大匹配数假设为x,那么有
n-x个王子没有娶到心仪的公主,有m-x个公主没被娶。那么,我们就虚拟出m-x个王子,他们喜欢所有的公主,再虚拟出n-x个公主,所有的王子都喜欢她们。这样王子公主都是有n+m-x个,并且是个完美匹配。
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N = 2e3+10;
const int M =2e5+10;
struct node
{
int to,nxt;
}g[M];
int head[N],cnt;
int link[N],match[N];
int dfn[N],low[N],id,sta[N],top,color[N],cl;
bool vis[N],mp[510][510];
int n,m,nn;
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;
memset(dfn,0,sizeof(dfn));
memset(head,-1,sizeof(head));
memset(mp,0,sizeof(mp));
}
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;
}
}
bool dfs(int u)
{
int v;
for(int i=head[u];i!=-1;i=g[i].nxt)
{
v=g[i].to;
if(vis[v]) continue;
vis[v]=1;
if(match[v]==-1||dfs(match[v]))
{
link[u]=v;
match[v]=u;
return 1;
}
}
return 0;
}
void hungary()
{
memset(match,-1,sizeof(match));
memset(link,0,sizeof(link));
for(int i=1;i<=n;i++)
{
memset(vis,0,sizeof(vis));
dfs(i);
}
}
void build()
{
int tot=2*nn;
for(int i=1;i<=n;i++)
{
if(!link[i])
{
link[i]=++tot;
match[tot]=i;
for(int j=1;j<=n;j++)
add(j,tot);
}
}
for(int i=1;i<=m;i++)
{
if(match[i+nn]==-1)
{
match[i+nn]=++tot;
link[tot]=i+nn;
for(int j=1;j<=m;j++)
add(tot,j+nn);
}
}
for(int i=1;i<=tot;i++)
if(link[i])
{
add(link[i],i);
}
memset(vis,0,sizeof(vis));
for(int i=1;i<=tot;i++)
if(!dfn[i]) tarjan(i);
}
int main()
{
int u,v,t,tt=0;
t=Read();
while(t--)
{
n=Read(),m=Read();
Init();
nn=max(n,m);
for(u=1;u<=n;u++)
{
x=Read();
while(x--)
{
v=Read();
mp[u][v]=1;
add(u,v+nn);
}
}
hungary();
build();
printf("Case #"); Write(++tt); puts(":");
for(u=1;u<=n;u++)
{
num=0;
for(int i=1;i<=m;i++)
{
if(mp[u][i]&&color[u]==color[nn+i])
ans[++num]=i;
}
Write(num);
for(int i=1;i<=num;i++)
{
putchar(' ');
Write(ans[i]);
}
putchar('\n');
}
}
return 0;
}