POJ 1149 最大流

我们可以把每位客人当作节点建立网络流，猪圈中猪的数量便是对流量的限制。首先，如果一个客人第一次打开a猪圈，那我们就在起点s与这位客人之间连一条容量为猪圈a中初始值得边（因为他是第一个开这个猪圈的人，无法调配），对于后来的每个客人，如果他们也想打开猪圈a，就可以从第一位客人这里连一条容量为无穷大的边（因为第一次开猪圈之后就可以调配了），然后求最大流。

#include<iostream>
#include<math.h>
#include<iomanip>
#include <string>
#include <cstdio>
#include<stdio.h>
#include <cstring>
#include <algorithm>
#include <queue>
#include<vector>
#define INF 0x3f3f3f3f
#define N 2010
#define ll long long
using namespace std;int stop;
int s,t;
struct Edge
{
    int Next,to,w;
    Edge(int a = 0,int b = 0,int c = 0){
        Next = a,to = b,w = c;
    }
}edge[N*10];
int cnt;
int head[N];
void addedge(int from,int to,int w){
    edge[cnt] = Edge(head[from],to,w);
    head[from] = cnt;cnt++;
    edge[cnt] = Edge(head[to],from,0);
    head[to] = cnt;cnt++;
}

int deep[N],cur[N];

bool bfs()
{
    memset(deep,0,sizeof(deep));
    deep[s]=1;
    queue<int>que;que.push(s);
    while(!que.empty())
    {
        int u=que.front();que.pop();
        for(int i=head[u];i!=-1;i=edge[i].Next)
        {
            int v=edge[i].to;
            if(!deep[v]&&edge[i].w>0)
            {
                deep[v]=deep[u]+1;que.push(v);
            }
        }
    }
    return deep[t];
}

int dfs(int u,int a)
{
    int flow = 0,f;
    if(u==t || a == 0)
        return a;
    for(int &i=cur[u];i!=-1;i=edge[i].Next)
    {
        int v=edge[i].to;
        if(deep[v]==deep[u]+1 && edge[i].w>0 && (f=dfs(v,min(a,edge[i].w))) > 0)
        {
            edge[i].w-=f;
            edge[i^1].w+=f;
            flow+=f;
            a-=f;
            if(a == 0)
                break;
        }
    }
    return flow;
}


int m,n;
int maxflow(){
    int flow = 0;
    while (bfs())
    {
        for(int i=0;i<=2*n;i++){
            cur[i]=head[i];
        }
        flow += dfs(s,INF);
    }
    return flow;
}


int zhu[N];
int belong[N];

int main()
{
    int x,y,w;
    int k;
    int d;
    memset(head,-1,sizeof(head));
    scanf("%d%d",&m,&n);
    s=0;t=n+1;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&zhu[i]);
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&k);
        y=0;
        for(int j=1;j<=k;j++)
        {
            scanf("%d",&x);
            if(belong[x]==0)
            {
                addedge(s,i,zhu[x]);
                belong[x]=i;
            }
            else
            {
                addedge(belong[x],i,INF);
            }
        }
        
        scanf("%d",&d);
        addedge(i,t,d);
    }
    printf("%d\n",maxflow());
    return 0;
}