HDU 4292 Food (成都赛区网络赛第五题,拆点网络流)

 

题意：跟POJ3281基本上一样的拆点网络流。

建图：建一超级源点和一超级汇点，源点与食物相连，边权为其数量，汇点与饮料相连，边权也为其数量，把人分成两个点，之间的边权为1。每个人与之需要的食物和饮料相连，边权为1。

代码(SAP模板)：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
const int maxn=101050;
const int maxm=3000000;
const int oo=1<<30;
int idx,N,F,D;
int cur[maxn],pre[maxn];
int dis[maxn],gap[maxn];
int aug[maxn],head[maxn];
struct Node
{
    int u, v, w;
    int next;
}edge[maxm];
void addEdge(int u, int v, int w)
{
    edge[idx].u=u;
    edge[idx].v=v;
    edge[idx].w=w;
    edge[idx].next=head[u];
    head[u]=idx++;
    edge[idx].u=v;
    edge[idx].v=u;
    edge[idx].w=0;
    edge[idx].next=head[v];
    head[v]=idx++;
}
int SAP(int s,int e,int n)
{
    int max_flow=0,v,u=s;
    int id,mindis;
    aug[s]=oo;
    pre[s]=-1;
    memset(dis,0,sizeof(dis));
    memset(gap,0,sizeof(gap));
    gap[0] = n;
    for (int i=0;i<=n;i++)
    {
        cur[i]=head[i];
    }
    while(dis[s]<n)
    {
        bool flag=false;
        if(u==e)
        {
            max_flow+=aug[e];
            for (v=pre[e];v!=-1;v=pre[v])
            {
                id = cur[v];
                edge[id].w-=aug[e];
                edge[id^1].w+=aug[e];
                aug[v]-=aug[e];
                if (edge[id].w==0)
                u=v;
            }
        }
        for(id=cur[u];id!=-1;id=edge[id].next)
        {
            v=edge[id].v;
            if(edge[id].w>0&&dis[u]==dis[v]+1)
            {
                flag=true;
                pre[v]=u;
                cur[u]=id;
                aug[v]=min(aug[u], edge[id].w);
                u=v;
                break;
            }
        }
        if (flag==false)
        {
            if(--gap[dis[u]]==0)
            break;
            mindis=n;
            cur[u]=head[u];
            for(id=head[u];id!=-1;id=edge[id].next)
            {
                v=edge[id].v;
                if(edge[id].w>0&&dis[v]<mindis)
                {
                    mindis=dis[v];
                    cur[u]=id;
                }
            }
            dis[u]=mindis+1;
            gap[dis[u]]++;
            if(u!=s)
            u=pre[u];
        }
    }
    return max_flow;
}
int main()
{
    while(~scanf("%d%d%d",&N,&F,&D))
    {
        int source=0,sink=N+N+F+D+10;
        memset(head,-1,sizeof(head));
        idx=0;
        int a,b,f,d;
        int food [10000];
        int drink[10000];
        char str[500][500];
        for(int i=1;i<=F;i++)
        {
            scanf("%d",&food[i]);
            addEdge(source,i,food[i]);
        }
        for(int i=1;i<=D;i++)
        {
            scanf("%d",&drink[i]);
            addEdge(i+F,sink,drink[i]);
        }
        for(int i=1;i<=N;i++)
        {
            addEdge(D+F+i,D+F+N+i,1);
        }
        for(int i=1;i<=N;i++)
        {
            scanf("%s",str[i]+1);
            for(int j=1;j<=F;j++)
            {
                if(str[i][j]=='Y')
                {
                    addEdge(j,F+D+i,1);
                }
            }
        }

        for(int i=1;i<=N;i++)
        {
            scanf("%s",str[i]+1);
            for(int j=1;j<=D;j++)
            {
                if(str[i][j]=='Y')
                {
                    addEdge(F+D+N+i,F+j,1);
                }
            }
        }
        int n=sink+1;
        printf("%d\n",SAP(source,sink,n));
    }
    return 0;
}