COGS 14. [网络流24题] 搭配飞行员

14. [网络流24题] 搭配飞行员

建边的时候要先从源点向主飞行员和从副飞行员向汇点连边

输入的时候只在两人之间连边

#include<cstdio>
#include<algorithm>
using namespace std;
#define maxn 10000
#define inf 100000000
int n,m,src,dec,cur[maxn],ans,lev[maxn];
int front[maxn],head,tail,que[maxn],tot;

struct node
{
    int to,next,cap;
}e[maxn];
inline void add(int u,int v,int w)
{
    e[++tot].to=v; e[tot].next=front[u]; e[tot].cap=w; front[u]=tot;
    e[++tot].to=u; e[tot].next=front[v]; e[tot].cap=0; front[v]=tot;
}

bool bfs()
{
    for(int i=src;i<=dec;i++) lev[i]=-1,cur[i]=front[i];
    head=tail=0; 
    que[tail++]=src; lev[src]=0;
    while(head<tail)
    {
        for(int i=front[que[head]];i;i=e[i].next)
            if(e[i].cap>0&&lev[e[i].to]==-1)
            {
                lev[e[i].to]=lev[que[head]]+1; 
                que[tail++]=e[i].to;
                if(e[i].to==dec) return true;
            }
        head++;
    }
    return false;
}
int dinic(int u,int flow)
{
    if(u==dec) return flow;
    int res=0,delta;
    for(int &i=cur[u];i;i=e[i].next)
    {
        if(e[i].cap>0&&lev[e[i].to]>lev[u])
        {
            delta=dinic(e[i].to,min(e[i].cap,flow-res));
            if(delta)
            {
                e[i].cap-=delta; e[i^1].cap+=delta;
                res+=delta; if(res==flow) break; 
            }
        }
    }
    if(res!=flow) lev[u]=-1;
    return res;
}

int main()
{
    freopen("flyer.in","r",stdin);
    freopen("flyer.out","w",stdout);
    scanf("%d%d",&n,&m);
    int u,v,w;
    src=0; dec=n+1;
    for(int i=1;i<=m;i++) add(src,i,1);
    for(int i=m+1;i<=n;i++) add(i,dec,1);
    while(scanf("%d%d",&u,&v)!=EOF)
    {
        add(u,v,1);
    }
    while(bfs())
        ans+=dinic(src,inf);
    printf("%d
",ans);
    return 0;
}