H - Food HDU - 4292

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=4292

读题

给定N个人、F种食物、M种饮料以及每种食物数量、每种饮料数量。 
通过一个N*F的矩阵给出某个人是否接受某种食物。 
再通过一个N*M矩阵给出某个人是否接受某种饮料。 
顾客的要求是同时有饮料和食物（即其接受的饮料至少提供一瓶，接受的食物至少提供一份）。 
求最多能满足多少位顾客。

解题

顾客同时需要满足饮料和食物两种需求，可以引一条从食物到顾客的有向弧、一条顾客到饮料的有向弧。那么每一条可行流都对应一个被满足的顾客，但不是一一对应。要满足一一对应关系，即顾客结点的结点容量为1，需要将顾客进行拆点。顾客u拆成u1和u2，u1、u2之间连一条容量为u结点容量的有向弧。原指向u的边改成指向u1，原从u指出的边改成从u2指出。 
如此，将饮料看作源点、食物看作汇点，这就是一个多源多汇模型。每条流量对应一个被满足的顾客。 
解决多源多汇模型，设立一个超级源点S，引入从S到源点的有向弧，容量为1.设立一个超级汇点T，引入汇点到超级汇点的有向弧，容量为1. 
超级源点到超级汇点的最大流即为答案。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
int const inf =0x3f3f3f3f;
int const maxe=1e5+1000;
int const maxv =1650;
using namespace std;
int N,F,D;
int S,T;//源点和汇点
struct edge
{
    int to, w, next;
} e[maxe<<1];
int head[maxv<<1],depth[maxv],cnt;
void init ()
{
    memset(head,-1,sizeof(head));
    cnt=-1;
}
void add_edge(int u,int to,int w)
{
    e[++cnt].w=w;
    e[cnt].to=to;
    e[cnt].next=head[u];
    head[u]=cnt;
}
void _add(int u,int v,int w)
{
    add_edge(u,v,w);
    add_edge(v,u,0);
}
bool bfs()
{
    queue<int>q;
    while(!q.empty()) q.pop();
    memset(depth,0,sizeof(depth));
    depth[S]=1;
    q.push(S);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u]; i!=-1; i=e[i].next)
        {
            if(!depth[e[i].to]&&e[i].w>0)
            {
                depth[e[i].to]=depth[u]+1;
                q.push(e[i].to);
            }
        }
    }
    if(!depth[T]) return false;
    return true;
}
int dfs(int u,int flow)
{
    if(u==T) return flow;
    int ret =0;
    for(int i=head[u]; i!=-1&& flow; i=e[i].next)
    {
        if(depth[u]+1==depth[e[i].to]&& e[i].w!=0)
        {
            int temp=dfs(e[i].to,min(e[i].w,flow));
            if(temp>0)
            {
                flow=flow-temp;
                ret=ret+temp;
                e[i].w=e[i].w-temp;
                e[i^1].w=e[i^1].w+temp;
            }
        }
    }
    return ret;
}
int Dinic()
{
    int ans=0;
    while(bfs())
        ans=ans+dfs(S,inf);
    printf("%d\n",ans);
}
char A[300];
int main()
{
    while(scanf("%d %d %d",&N,&F,&D)!=EOF)
    {
        S=0,T=1600;
        init();
        for(int i=1; i<=F; i++)
        {
            int temp;
            scanf("%d",&temp);
            _add(i,i+250,temp);
            _add(S,i,inf);
        }
        for(int i=1; i<=D; i++)
        {
            int temp;
            scanf("%d",&temp);
            _add(i+500,i+750,temp);
            _add(i+750,T,inf);
        }
        for(int i=1; i<=N; i++)
            _add(i+1000,i+1250,1);
        for(int i=1; i<=N; i++)
        {
            scanf("%s",A);
            for(int j=0; A[j]!='\0'; j++)
            {
                if(A[j]=='N') continue;
                else
                    _add(j+251,i+1000,1);
            }
        }
        for(int i=1; i<=N; i++)
        {
            scanf("%s",A);
            for(int j=0; A[j]!='\0'; j++)
            {
                if(A[j]=='N') continue;
                else
                    _add(i+1250,j+501,1);
            }
        }
          Dinic();
    }
    return 0;
}