hdu 4685 简单匹配+Tarjan算法

思路:首先看到这题以为能用poj1904的模版直接A掉,WA了几次,然后又TLE了几次。还是想到了正解。

一开始我想的大致方向已经是对的了。先是由王子向每个喜欢的公主建边,再求一次最大匹配,找出匹配后,由匹配的公主向王子建边。

但可能会有没有匹配到的公主和王子,那么这个王子可以和任何它喜欢的公主结婚,这个公主也可以和任何喜欢她的王子结婚。

因为这些不在匹配中的点,加到匹配中后,减少的匹配数和增加的匹配数都是1。

我们也就想像poj1904那样,将他们变为一个强连通分量,我开始出错就在这。

直接在原图上将他们建边变为强连通分量会使原图性质发生改变。

所有我们对每个没有匹配的公主,建一个虚拟的王子,让他们变成匹配,然后由这个虚拟王子向每个公主建边。

对每个没有匹配的王子,建一个虚拟的公主,让他们变成匹配,然后每个王子向这个虚拟公主建边。

求一个Tarjan,判断是否为1个强连通分量即可。

#include<iostream>

#include<cstdio>

#include<cstring>

#include<algorithm>

#include<cmath>

#include<vector>

#define Max(a,b) ((a)>(b)?(a):(b))

#define Min(a,b) ((a)<(b)?(a):(b))

#define Maxn 2010

#define Maxm 400010

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[Maxm];

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,j,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;

}

 

 

你可能感兴趣的:(tar)