POJ1149 最大流(Isap)

题意： 有 M 个猪圈，每个猪圈里初始时有若干头猪。一开始所有猪圈都是关闭的。依次来了 N 个顾客，每个顾客分别会打开指定的几个猪圈，从中买若干头猪。每个顾客分别都有他能够买的数量的上限。每个顾客走后，他打开的那些猪圈中的猪，都可以被任意地调换到其它开着的猪圈里，然后所有猪圈重新关上。

构图：

(1) 将顾客看作源点和汇点以外的结点，并设一个源点和汇点

(2)源点和每个猪圈的第一个顾客连边，边权是开始时猪圈中猪的数目

(3)顾客j在顾客i之后打开猪圈，i，j 之间连一条权值为inf的边，因为迈克可以根据j的需要，从其他猪圈调猪过来，让j得到尽可能多的猪

(4)每个顾客和汇点连一条权值为顾客要买猪的数目 的边

(5)然后就是套模板了。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
using namespace std;
#define inf 0x7fffffff
#define N 105
#define M 10005
int size,head[N];
int cur[N],pre[N],dis[N],gap[N];
int last[1005],house[1005];
struct Edge
{
    int v,w,next;
    Edge(){}
    Edge(int V,int W,int NEXT):v(V),w(W),next(NEXT){}
}edge[M];
void Init() //初始化 
{
    memset(head,-1,sizeof(head));
    memset(last,-1,sizeof(last));
    size = 0;
}
void InsertEdge(int u,int v,int w) //加边 
{
    edge[size] = Edge(v,w,head[u]);
    head[u] = size++;
    edge[size] = Edge(u,0,head[v]);
    head[v] = size++;
}

int Isap(int st,int ed,int n)//n为结点个数，Isap模板  
{
    for(int i=0; i<=n; i++)
    {
        dis[i] = gap[i] = 0;
        cur[i] = head[i];
    }
    int u = pre[st] = st;
    int aug = inf , maxflow = 0;
    while(dis[st] < n)
    {
loop:
        for(int &i=cur[u]; i!=-1; i = edge[i].next)
        {
            int v = edge[i].v;
            if(edge[i].w && dis[u] == dis[v]+1)
            {
                aug = aug < edge[i].w ? aug:edge[i].w;
                pre[v] = u; 
                u = v;
                if(v == ed)
                {
                    maxflow += aug;
                    for( u = pre[u]; v!=st; v = u,u = pre[u])
                    {
                        edge[cur[u]].w -= aug;
                        edge[cur[u]^1].w += aug;
                    }
                    aug = inf;
                }
                goto loop;
            }
        }
        int mindis = n;
        for(int i = head[u]; i!=-1; i=edge[i].next)
        {
            int v = edge[i].v;
            if(edge[i].w && dis[v] < mindis)
            {
                cur[u] = i;
                mindis = dis[v];
            }
        }
        if(--gap[dis[u]]==0) break;
        gap[dis[u] = mindis+1]++;
        u = pre[u];
    }
    return maxflow;
}

int main()
{
    int m,n,num,k,st,ed,nv;
    scanf("%d%d",&m,&n);
    Init();
    st = 0; // 附加源点 
    ed = n+1; //附加汇点 
    nv = n+2; //结点个数 
    for(int i=1; i<=m; i++)
    scanf("%d",&house[i]);
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&num);
        for(int j=0; j<num; j++)
        {
            scanf("%d",&k);
            if(last[k]!=-1) //last数组记录k是不是第一次打开 
            InsertEdge(last[k],i,inf); //(3) 
            else 
            {
                InsertEdge(st,i,house[k]); //(2)
                last[k] = i;
            }
        }
        scanf("%d",&k);
        InsertEdge(i,ed,k); //(4)
    }
    int ans = Isap(st,ed,nv);
    printf("%d
",ans);
    return 0;
}