hdu 4292 Food 最大流

将人拆点,之间连接一条容量为1的边,表示一个人只能选一次,从s到每种食物连接一条容量为该种食物总数的边,从每种饮料连接一条容量为该种饮料总量的边,如果一个人喜欢某种食物或饮料,连边即可。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=1200;
const int inf=1000000;
int n,f,d;
int level[maxn],que[maxn];
int head[maxn],lon;
int min(int a,int b)
{
    if(a<b) return a;
    else return b;
}
struct
{
    int next,to,c;
}e[maxn*1000];
void edgeini()
{
    memset(head,-1,sizeof(head));
    lon=-1;
}
void edgemake(int from,int to,int c)
{
    e[++lon].c=c;
    e[lon].to=to;
    e[lon].next=head[from];
    head[from]=lon;
}
void make(int from,int to,int c)
{
    edgemake(from,to,c);
    edgemake(to,from,0);
}

bool makelevel(int s,int t)
{
    memset(level,0,sizeof(level));
    int front=1,end=0;
    que[++end]=s;
    level[s]=1;
    while(front<=end)
    {
        int u=que[front++];
        if(u==t) return true;
        for(int k=head[u];k!=-1;k=e[k].next)
        {
            int v=e[k].to;
            if(!level[v]&&e[k].c)
            {
                que[++end]=v;
                level[v]=level[u]+1;
            }
        }
    }
    return false;
}

int dfs(int now,int t,int maxf)
{
    if(now==t) return maxf;
    int ret=0;
    for(int k=head[now];k!=-1;k=e[k].next)
    {
        int u=e[k].to;
        if(level[u]==level[now]+1&&e[k].c)
        {
            int f=dfs(u,t,min(e[k].c,maxf-ret));
            e[k].c-=f;
            e[k^1].c+=f;
            ret+=f;
            if(ret==maxf) return ret;
        }
    }
    if(ret==0) level[now]=0;
    return ret;
}

int maxflow(int s,int t)
{
    int ret=0;
    while(makelevel(s,t))
    {
        ret+=dfs(s,t,inf);
    }
    return ret;
}
int main()
{
    int i,j;
    while(scanf("%d%d%d",&n,&f,&d)!=EOF)
    {
        edgeini();
        int t=n+n+f+d+1;
        int x;
        char y;
        for(i=1;i<=f;i++)
        {
            scanf("%d",&x);
            make(0,i,x);
        }
        for(i=1;i<=n;i++) make(f+i,f+n+i,1);
        for(i=1;i<=d;i++)
        {
            scanf("%d",&x);
            make(f+n+n+i,t,x);
        }
        getchar();
        for(i=1;i<=n;i++)
        {    
            for(j=1;j<=f;j++)
            {
                scanf("%c",&y);
                if(y=='Y')
                    make(j,i+f,1);
            }
            getchar();
        }
        for(i=1;i<=n;i++)
        {    
            for(j=1;j<=d;j++)
            {
                scanf("%c",&y);
                if(y=='Y')
                    make(i+f+n,j+f+n+n,1);
            }
            getchar();
        }
        int ans=maxflow(0,t);
        printf("%d
",ans);
    }
    return 0;
}
    


 

原文地址:https://www.cnblogs.com/vermouth/p/3710177.html