Poj 1149 PIGS

题解：有M个猪圈。有N个游客，每个游客想买一定数量的猪，并且有一些猪圈的钥匙。每一个顾客走之后，他打开的猪圈的猪可以分配到其它已打开的猪圈中。求最多能卖出多少猪。

题解：参考博文：http://www.cnblogs.com/sleeper-qp/archive/2012/07/23/2605253.html

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <string>
#include <vector>
#include <list>
#include <map>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <numeric>
#include <functional>
#include <set>
#include <fstream>

using namespace std;

const int INF=0x3f3f3f3f;
const int maxm=1010;
const int maxn=110;

int N,M;
int customer[maxm][maxn];
int cusnum[maxm];
int pighouse[maxm];
int B[maxn];
struct edge{
    int u,v,cap,next;
}G[maxn*maxm];
int head[maxn*maxm],pre[maxn*maxm],level[maxn*maxm];
int num[maxn*maxm],cur[maxn*maxm];
int idx;
int s,t,tt;

void build(int u,int v,int cap)
{
    G[idx].v=v;
    G[idx].cap=cap;
    G[idx].next=head[u];
    head[u]=idx++;
}

void add_edge(int u,int v,int cap)
{
    build(u,v,cap);
    build(v,u,0);
}

void bfs()
{
    memset(level,-1,sizeof(level));
    memset(num,0,sizeof(num));
    queue<int> q;
    q.push(t);
    level[t]=0;
    num[0]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u];i!=-1;i=G[i].next)
        {
            int v=G[i].v;
            if(level[v]==-1){
                level[v]=level[u]+1;
                num[level[v]]++;
                q.push(v);
            }
        }
    }
}

void ISAP()
{
    memcpy(cur,head,sizeof(cur));
    bfs();
    int flow=0;
    int u=pre[s]=s;
    while(level[s]<tt)
    {
        if(u==t){
            int f=INF,pos;
            for(int i=s;i!=t;i=G[cur[i]].v)
            {
                if(f>G[cur[i]].cap){
                    f=G[cur[i]].cap;
                    pos=i;
                }
            }
            for(int i=s;i!=t;i=G[cur[i]].v)
            {
                G[cur[i]].cap-=f;
                G[cur[i]^1].cap+=f;
            }
            flow+=f;
            u=pos;
        }
        int k;
        for(k=cur[u];k!=-1;k=G[k].next)
        {
            if(level[G[k].v]+1==level[u]&&G[k].cap) break;
        }
        if(k!=-1){
            cur[u]=k;
            pre[G[k].v]=u;
            u=G[k].v;
        }else{
            if(--num[level[u]]==0) break;
            int mind=tt;
            for(int i=head[u];i!=-1;i=G[i].next)
            {
                if(mind>level[G[i].v]&&G[i].cap){
                    mind=level[G[i].v];
                    cur[u]=i;
                }
            }
            level[u]=mind+1;
            num[level[u]]++;
            u=pre[u];
        }
    }
    printf("%d
",flow);
}

void init()
{
    memset(cusnum,0,sizeof(cusnum));
    memset(head,-1,sizeof(head));
    //    memset(timec,0,sizeof(timec));
    s=0;
    t=N+1;
    tt=t+1;
    idx=0;
}

void build_graph()
{
    for(int i=1;i<=N;i++)
    {
        add_edge(i,t,B[i]);
    }
    for(int i=1;i<=M;i++)
    {
        if(cusnum[i]>0){
            add_edge(s,customer[i][0],pighouse[i]);
        }
        for(int j=1;j<cusnum[i];j++)
        {
            add_edge(customer[i][j-1],customer[i][j],INF);
        }
    }
}

int main()
{
//    freopen("/Users/apple/Desktop/暑假/POJ 1149 PIGS/POJ 1149 PIGS/in","r",stdin);
    while(scanf("%d%d",&M,&N)!=EOF)
    {
        init();
        for(int i=1;i<=M;i++)
        {
            scanf("%d",&pighouse[i]);
        }
        for(int i=1;i<=N;i++)
        {
            int A,key;
            scanf("%d",&A);
            for(int j=1;j<=A;j++)
            {
                scanf("%d",&key);
                customer[key][cusnum[key]++]=i;
            }
            scanf("%d",&B[i]);
        }
        // puts("YES");
        build_graph();
        // puts("YES");
        ISAP();
        // puts("YES");
    }
    return 0;
}