POJ 1904 King's Quest 强连通分量+二分图
题意:有n个王子和n个女孩,一位王子可以喜欢多个女孩,但是只能与一个女孩结婚,首先,通过题目中给出的王子的编号与王子喜欢的女孩的编号,可以建立一个图(王子指向女孩),这个图只是简单记录了哪位王子喜欢哪些女孩;然后通过二分图将王子与这些女孩进行匹配,找出王子可以结婚的女孩,这个时候,一个王子只能有一个女孩了(即结婚对象),加一条反向边(女孩指向王子),这代表着王子的结婚对象;
通过以上操作,就建立了一个有向图了,这个时候利用tarjan算法找出强连通分量即可,输出每一个连通分量中的个数与女孩的编号。
注意:王子的编号是从1开始的,女孩的编号也是从1开始的,这样会有重复,会有混乱,所以让女孩的编号从n+1到2*n会比较好。
强连通分量通过的代码
自己的错误代码
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<stack>
#include<string.h>
using namespace std;
const int maxn=4050;
int dfn[maxn],low[maxn],Stack[maxn],color[maxn],vis[maxn],sum[maxn],ans[maxn];
int match[maxn],vis1[maxn],girl[maxn];
int n,index,len,cnt;
vector< vector<int> >g;
void tarjan(int u)
{
dfn[u]=low[u]=++index;
Stack[cnt++]=u;
vis[u]=1;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(!dfn[v])
{
tarjan(v);
low[u]=min(low[u],low[v]);
}else if(vis[v])
{
low[u]=min(dfn[v],low[u]);
}
}
if(dfn[u]==low[u])
{
++len;
int LL=cnt;
int v;
do{
v=Stack[--cnt];
color[v]=len;
vis[v]=0;
}while(u!=v);
sum[len]=LL-cnt;
}
}
bool dfs(int k)
{
for(int j=0;j<g[k].size();j++)
{
int v=g[k][j];
if(!vis[v])
{
vis[v]=1;
if(match[v]==-1 || dfs(match[v]))
{
match[v]=k;
return true;
}
}
}
}
int Hunary()
{
int all=0;
memset(match,0xff,sizeof(match));
for(int i=1;i<=n;i++)
{
memset(vis1,0,sizeof(vis1));
if(dfs(i))
all++;
}
return all;
}
void init()
{
memset(dfn,0,sizeof(dfn));
memset(low,0,sizeof(low));
memset(Stack,0,sizeof(Stack));
memset(color,0,sizeof(color));
memset(vis,0,sizeof(vis));
g.clear();
g.resize(maxn+1);
index=len=cnt=0;
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
init();
int k,x;
for(int i=1;i<=n;i++)
{
scanf("%d",&k);
for(int j=1;j<=k;j++)
{
scanf("%d",&x);
g[i].push_back(x+n);
}
}
for(int i=1;i<=n;i++)
{
scanf("%d",&x);
girl[i]=x+n;
}
int q=Hunary();
for(int i=1;i<=2*n;i++)
{
if(match[i]!=-1)
g[i].push_back(match[i]);
}
for(int i=1;i<=n;i++)
{
if(!dfn[i])
tarjan(i);
}
for(int i=1;i<=n;i++)
{
memset(ans,0,sizeof(ans));
int length=g[i].size();
int u=color[i],v;
int m=0;
for(int j=0;j<length;j++)
{
v=color[ g[i][j] ];
if(u==v)
{
ans[m++]=g[i][j];
}
}
printf("%d",m);
sort(ans,ans+m);
for(int j=0;j<m;j++)
{
printf(" %d",ans[j]-n);
}
printf("\n");
}
}
return 0;
}
#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<vector>
using namespace std;
const int maxn=2222*2222;
int n;
int dfn[maxn],low[maxn],c[maxn],f[maxn],girl[maxn],sum[maxn];
int match[maxn],vis[maxn];
int l=0,cnt=0,start=0;
vector<int>g[maxn];
void tarjan(int u)
{
dfn[u]=low[u]=++start;
c[++l]=u;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(!dfn[v])
{
tarjan(v);
low[u]=min(low[u],low[v]);
}else if(!f[v])
{
low[u]=min(low[u],dfn[v]);
}
}
if(dfn[u]==low[u])
{
int len=l;
cnt++;
while(c[l]!=u)
{
f[c[l--]]=cnt;
}
f[c[l--]]=cnt;
sum[cnt]=len-l;
}
}
bool dfs(int k)
{
for(int j=0;j<g[k].size();j++)
{
int v=g[k][j];
if(!vis[v])
{
vis[v]=1;
if(match[v]==-1 || dfs(match[v]))
{
match[v]=k;
return true;
}
}
}
return false;
}
int Hunary()
{
int all=0;
memset(match,0xff,sizeof(match));
for(int i=1;i<=n;i++)
{
memset(vis,0,sizeof(vis));
if(dfs(i))
all++;
}
return all;
}
int main()
{
scanf("%d",&n);
int p=n;
for(int i=1;i<=n;i++)
{
int k;
scanf("%d",&k);
for(int j=1;j<=k;j++)
{
int x;
scanf("%d",&x);
// x+=p;
g[i].push_back(x);
}
}
for(int i=1;i<=n;i++)
{
int y;
scanf("%d",&y);
girl[i]=y+n;
}
int ans=Hunary();
printf("ans=%d\n",ans);
//for(int i=1;i<=n;i++)
//{
// printf("i= %d mach[i]=%d ",i,match[i]);
// match[i]=match[i];
// printf("i= %d mach[i]=%d \n",i,match[i]);
//}
for(int i=1;i<=n;i++)
{
for(int j=0;j<g[i].size();j++)
{
int v=g[i][j];
if(match[i]==v)
{
g[v].push_back(i)
}
}
}
memset(dfn,0,sizeof(dfn));
for(int i=1;i<=n;i++)
{
if(!dfn[i])
tarjan(i);
}
for(int i=1;i<=n;i++)
{
printf("%d ",sum[i]);
for(int j=0;j<g[i].size();j++)
{
int v=g[i][j];
if(f[i]==f[v])
{
printf("%d ",g[i][j]);
}
}
printf("\n");
}
return 0;
}