uva 753 A Plug for UNIX

最大流 OR 二分图匹配

题意:输入n,有n个插座,下面n行是每个插座的类型(最后24个字母来表示一个插座,没有空格放心用scanf,但是有可能插座会相同,但是这个没有什么影响)

         输入m,有m个电器,下面m行每行两个单词分别是电器的名字和插头类型(同样24个字母单词内没空格,两个单词空格隔开)

         输入k,有k个转换器,下面k行每行两个单词,分别表示转换器的入口类型和插头类型

每种转换器的个数是无限的,转换器本身可以与转换器相连

要你求,让最多的电器能够插在插座上(可以用转换器辅助也可以直接插上去),输入不能插上去的电器的数量

 

思路一:转化为最大流,重点还是如何建图,我是用邻接表建图的这样效率也高一点,其实用邻接矩阵建图也可以。我的建图方法是,电器从1到m标号,转换器从m+1到m+k标号,插座从m+k+1到m+k+n标号,另外设置一个源点s=0,汇点t=m+k+n+1 , 求s到t的最大流

s与所有电器建一条有向边,容量为1,所有插座与汇点建一条有向边,容量为1。对于每个电器,如果能直接和插座相连的,和每个能相连的插座建一条有向边,容量为1.另外,所有电器和所有能连接上的转换器建一条有向边,容量为INF,转换器和转换器之间能相连的建一条有向边容量为INF,转换器和插座能相连建一条有向边容量为INF。

建图完毕,直接EK最大流模板上去即可

注意几点:

1.邻接表建有向边不要忘记每条有向边附带一条反边,反边的容量为0

2.一定要记得建立电器和插座直接相连的边

3.一开始想复杂了,认为用电器和插座要拆点,转换器不用拆点,其实全部都不用拆点

4.不需要建无向图,另外建无向图应该是错的

#include <cstdio>

#include <cstring>

#include <queue>

using namespace std;

#define INF 0x3f3f3f3f

#define MAXN 550

#define MAXM 50010

int N,M,K,CNT,edgenum,s,t,first[MAXN];

struct edge

{

    int u,v,f,cap,next;

}e[MAXM];

struct  device  //电器

{

    char s1[30],s2[30];

}d[MAXN];

struct adapter //转换器

{

    char s1[30], s2[30];

}a[MAXN];

struct receptacle  //插座

{

    char s[30];

}r[MAXN]; 





void printfff()  //测试打印函数

{

    printf("M=%d  N=%d   K=%d  CNT=%d  s=%d   t=%d\n",M,N,K,CNT,s,t);

    for(int i=0; i<edgenum; i++)

        printf("%d:  %d-->%d=%d %d\n",i,e[i].u,e[i].v,e[i].cap,e[i].next);

}

void add(int u  ,int v , int cap)

{

    int t;

    e[edgenum].u=u; e[edgenum].v=v;

    e[edgenum].f=0; e[edgenum].cap=cap;

    e[edgenum].next=first[u];

    first[u]=edgenum++;



    t=u; u=v; v=t;



    e[edgenum].u=u; e[edgenum].v=v;

    e[edgenum].f=0; e[edgenum].cap=0;

    e[edgenum].next=first[u];

    first[u]=edgenum++;

}

void input()

{

    scanf("%d",&N);

    for(int i=1; i<=N; i++)

        scanf("%s",r[i].s);



    scanf("%d",&M);

    for(int i=1; i<=M; i++)

        scanf("%s%s",d[i].s1,d[i].s2);



    scanf("%d",&K);

    for(int i=1; i<=K; i++)

        scanf("%s%s",a[i].s1,a[i].s2); 



    CNT=N+M+K;

}

void init()

{

    memset(first,-1,sizeof(first));

    edgenum=0;

    s=0;  t=CNT+1; //源点,汇点

    for(int i=1; i<=M; i++)  //所有电器

    {

        add(s,i,1);  //源点和电器建边容量为1

        

        for(int j=1; j<=K ; j++) //所有电器和转换器的连接

            if(!strcmp(d[i].s2 , a[j].s1)) //电器可以插入转换器中

                add(i,j+M,INF);



        for(int j=1; j<=N; j++)  //电器和插座直接相连

            if(!strcmp(d[i].s2 , r[j].s))

                add(i,M+K+j,1);

    }



    for(int i=1; i<=K; i++) 

    {

        for(int j=1; j<=K; j++) //所有转换器和转换器之间的连接

            if(i!=j && !strcmp(a[i].s2 , a[j].s1))  

                //不用考虑相同种类的转换器的连接,因为没有意义

                add(i+M,j+M,INF);



        for(int j=1; j<=N; j++) //所有转换器和所有插座连接

            if(!strcmp(a[i].s2 , r[j].s)) //转换器可插入插座中

                add(i+M , M+K+j , INF);  //转化器和插座相连

    }



    for(int i=1; i<=N; i++)  //插座和汇点相连

        add(M+K+i,t,1);

}

void EK()

{

    queue<int>q;

    int a[MAXN],path[MAXN];

    int F;

    F=0;

    while(1)

    {

        memset(path,-1,sizeof(path));

        memset(a,0,sizeof(a));

        a[s]=INF;

        q.push(s);

        while(!q.empty())

        {

            int u=q.front(); q.pop();

            for(int k=first[u]; k!=-1; k=e[k].next)

            {

                int v=e[k].v;

                if(!a[v] && e[k].f<e[k].cap)

                {

                    a[v]=a[u]<e[k].cap-e[k].f?a[u]:e[k].cap-e[k].f;

                    path[v]=k;

                    q.push(v);

                }

            }

        }



        if(!a[t]) break;



        for(int k=path[t]; k!=-1; k=path[e[k].u]) //增广

        {

            //printf("%d<--%d ",e[k].v,e[k].u);

            e[k].f+=a[t];

            e[k^1].f-=a[t];

        }

        //printf("\n");

        F+=a[t];

    }



    //printf("F=%d\n",F);

    printf("%d\n",M-F);

}

int main()

{

    int T;

    scanf("%d",&T);

    while(T--)

    {

        input();

        init();

        //printfff();

        EK();

        if(T) printf("\n");

    }

    return 0;

}

 

 

 

思路二:二分图最大匹配

其实用二分图的思想来看这个问题更加直观,即电器和插座匹配,(电器之间,插座之间不可能匹配)。问题同样是建图问题

用最大流建图的话,顶点会有转换器,但是二分图来建则没有(当然也不需要加额外的源点和汇点)。二分图建立的分两部分,一部分是,用电器能不能直接和某些插座直接相连,能的话直接建边。另外要利用转换器,看看能不能把不能直接相连电器和插座给连接起来,能的话就连接。我还没写,但是想到是用dfs去找,不知道效率如何。

最后建了二分图,不需要管那些边是怎么连接的(直接相连或者借助了转换器),直接把匈牙利丢上去即可

明天起来再写二分图了

 

 

二分图写好了,思路就是上面说的那样,嗯写二分图舒心多了,建边少代码也少一些。不过我是用dfs去判断电器能不能通过转换器连到插座上的,所以时间慢了点,0.060s,上面的最大流是0.008s

另外一个地方是。电器不能直接与插座相连,然后用dfs去找能不能通过转换器去相连,虽然每种转换器的数量是无限的,但是在一趟寻找中一种转换器没必要被用两次(其实那种入口和插头相同的转换器也没必要用上,但是不做特殊判断了,没影响),也就是可以看成路径,虽然有环,但是这个环显然是没必要经过,所以加了一个vis数组标记

#include <cstdio>

#include <cstring>

#include <queue>

using namespace std;

#define INF 0x3f3f3f3f

#define MAXN 250

#define MAXM 40100

int N,M,K,CNT,edgenum,first[MAXN],mat[MAXN],vis[MAXN],cov[MAXN];

struct edge

{

    int u,v,next;

}e[MAXM];

struct  device  //电器

{

    char s1[30],s2[30];

}d[MAXN];

struct adapter //转换器

{

    char s1[30], s2[30];

}a[MAXN];

struct receptacle  //插座

{

    char s[30];

}r[MAXN]; 

void add(int u , int v)

{

    e[edgenum].u=u; e[edgenum].v=v;

    e[edgenum].next=first[u];

    first[u]=edgenum++;

}

int dfs(int j , int k)  //目标是j插座,当前是k转换器

{

    if(!strcmp(a[k].s2 , r[j].s))  //当前转换器能插入插座中

        return 1;

    vis[k]=1;  //标记k转换器被用过,在一趟dfs中一种转换器没必要被用两次

    for(int i=1; i<=K; i++) if(!vis[i]) //所有没被用过的转换器

        if(!strcmp(a[k].s2 , a[i].s1))         

        {//另外k转换器可以插入i转换器中

            if(dfs(j,i))

                return 1;

            vis[i]=0;

        }

    return 0;



}

int find(int i , int j)  //i电器和j插座

{

    memset(vis,0,sizeof(vis));

    for(int k=1; k<=K; k++) //所有转换器

        if(!strcmp(d[i].s2 , a[k].s1)) 

        {//电器能插入转换器的入口

            if(dfs(j,k))

                return 1;

            vis[k]=0;

        }

    return 0;

}

void input()

{

    scanf("%d",&N);

    for(int i=1; i<=N; i++)

        scanf("%s",r[i].s);



    scanf("%d",&M);

    for(int i=1; i<=M; i++)

        scanf("%s%s",d[i].s1,d[i].s2);



    scanf("%d",&K);

    for(int i=1; i<=K; i++)

        scanf("%s%s",a[i].s1,a[i].s2); 



    CNT=N+M+K;

}

void init()

{

    memset(first,-1,sizeof(first));

    edgenum=0;

    for(int i=1; i<=M; i++)  //所有电器 

        for(int j=1; j<=N; j++) //所有插座

        {

            if(!strcmp(d[i].s2 , r[j].s))  //可以直接相连

            {

                //printf("直接相连%d---%d\n",i,j);

                add(i,j+M);  //建立有向边

            }

            else if(find(i,j))  //如果通过转换器能连接上

            {

                //printf("通过转换器连接%d---%d\n",i,j);

                add(i,j+M);

            }

        }

}

void printfff()

{

    for(int i=0; i<edgenum; i++)

        printf("%d %d %d\n",e[i].u,e[i].v,e[i].next);

}

int dfs_match(int u)

{

    for(int k=first[u]; k!=-1; k=e[k].next)

    {

        int v=e[k].v;

        if(!cov[v])

        {

            cov[v]=1;

            if( mat[v]==-1 || dfs_match(mat[v]) )

            { mat[v]=u; return 1; }

        }

    }

    return 0;

}

void maxmatch()

{

    int max=0;

    memset(mat,-1,sizeof(mat));

    for(int i=1; i<=M+N; i++)  //所有的点都做一次起点

    {

        memset(cov,0,sizeof(cov));

        max+=dfs_match(i);

    }

    //for(int i=1; i<=M+N; i++)

        //printf("%d<--->%d\n",i,mat[i]);

    //printf("max=%d\n",max);

    printf("%d\n",M-max);

}

int main()

{

    int T;

    scanf("%d",&T);

    while(T--)

    {

        input();

        init();

       //printfff();

        maxmatch();

        if(T) printf("\n");

    }

    return 0;

}

 

 

你可能感兴趣的:(unix)