POJ-1149 PIGS (网络流巧妙建图)

题目链接

题意：

有M个猪圈，每个猪圈里初始时有若干头猪。一开始所有猪圈都是关闭的。依次来了N个顾客，每个顾客分别会打开指定的几个猪圈，从中买若干头猪。每个顾客分别都有他能够买的数量的上限。每个顾客走后，他打开的那些猪圈中的猪，都可以被任意地调换到其它开着的猪圈里，然后所有猪圈重新关上。问总共最多能卖出多少头猪。（1 <= N <= 100, 1 <= M <= 1000）

题解：

网络流巧妙建图。每个顾客分别用一个结点来表示。 对于每个猪圈的第一个顾客，从源点向他连一条边，容量就是该猪圈里的猪的初始数量。如果从源点到一名顾客有多条边，则可以把它们合并成一条，容量相加。 
对于每个猪圈，假设有n个顾客打开过它，则对所有整数i∈[1, n)，从该猪圈的第i个顾客向第i + 1个顾客连一条边，容量为∞。 
从各个顾客到汇点各有一条边，容量是各个顾客能买的数量上限。具体参见这里

代码：

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <bitset>
using namespace std;

typedef long long ll;
#define int ll
#define INF 0x3f3f3f3f3f3f3f3f
#define MAXM 1000 + 10
#define MAXN 100 + 10
const ll mod = 1e9 + 7;
#define P pair<int, int>
#define fir first
#define sec second

int m, n;
int num[MAXM], pre[MAXM];

int dep[MAXN];
struct Edge{
    int next, to, dis;
}edge[MAXM<<1];
int cnt, head[MAXN];
queue<int> q;

void add(int from, int to, int dis)
{
    edge[++cnt].next = head[from];
    edge[cnt].to = to;
    edge[cnt].dis = dis;
    head[from] = cnt;
}

bool bfs(int s, int t)
{
    memset(dep,0x7f,sizeof(dep));
    while (!q.empty()) q.pop();
    dep[s] = 0;
    q.push(s);

    while (!q.empty()) {
        int now = q.front(); q.pop();
        for (int i = head[now]; i != -1; i = edge[i].next) {
            if (dep[edge[i].to] > INF && edge[i].dis) {
                dep[edge[i].to] = dep[now]+1;
                q.push(edge[i].to);
            }
        }
    }
    if (dep[t] < INF) return true;
    else return false;
}

int dfs(int now, int t, int limit)
{
    if (!limit || now == t) return limit;

    int flow = 0, f;
    for (int i = head[now]; i != -1; i = edge[i].next) {
        if (dep[edge[i].to] == dep[now] + 1 && (f = dfs(edge[i].to, t, min(limit, edge[i].dis)))) {
            flow += f;
            limit -= f;
            edge[i].dis -= f;
            edge[i^1].dis += f;
            if (!limit) break;
        }
    }
    return flow;
}

int Dinic(int s, int t)
{
    int maxflow = 0;
    while (bfs(s, t)) maxflow += dfs(s, t, INF);
    return maxflow;
}

void init()
{
    cnt = -1;
    memset(head, -1, sizeof(head));
}

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);

    init();
    cin >> m >> n;
    int st = 0, ed = n + 1;
    for(int i = 1; i <= m; i ++) cin >> num[i];
    for(int i = 1; i <= n; i ++) {
        int k; cin >> k;
        for(int j = 1; j <= k; j ++) {
            int id; cin >> id;
            if(!pre[id]) {
                pre[id] = i;
                add(st, i, num[id]), add(i, st, 0);
            }
            else {
                add(pre[id], i, INF), add(i, pre[id], 0);
            }
        }
        int b; cin >> b;
        add(i, ed, b), add(ed, i, 0);
    }
    cout << Dinic(st, ed) << endl;
}

/*

老实说，
要到几岁才开始不相信圣诞老人的存在……
这类无聊的话题对我而言，
根本不痛不痒的。
不过，
讲到我从几岁起开始不相信圣诞老人就是那个穿着红衣服的老公公时，
我能确定地说，
我根本打从一开始就不相信。
我知道幼稚园圣诞节庆祝会时出现的圣诞老人是假的，
回溯记忆，
还能记起周围的幼稚园小朋友都一脸不信任地望着假扮圣诞老人的园长老师。
即使没有撞见老妈正在亲吻圣诞老公公，
机灵的我也老早就怀疑只在圣诞节工作的老爷爷是否真的存在了。
不过，
我却是过了很久以后，
才发现外星人、幽灵、妖怪、超能力者以及特摄、动画里头，
那些与邪恶组织战斗的英雄们并不存在这世上。
不，
说不定我早就发现了，
只不过一直不想承认而已。
因为，
在我的内心深处，
是十分渴望那些外星人、幽灵、妖怪、超能力者以及邪恶组织突然出现在眼前的。
和我生活的这个普通世界相比，
特摄、动画里头所描绘的世界，
反而更有魅力。
我也想活在那种世界里!
我真的好想拯救被外星人绑架并关在透明的大型豌豆夹里头的少女…
也想拿着雷射枪运用智慧与勇气击退企图改写历史的未来人…
或者光用一句咒语就收拾了恶灵跟妖怪，
再不然就是和秘密组织的超能力者进行超能力的战斗!
等等，
冷静一下，
假设我被外星人等等（以下略）那类的生物袭击，
没有任何特殊能力的我怎么可能和他们对抗?
于是，
我便如是幻想 ——
某天，
班上突然转来一个谜样的转学生，
他其实是个外星人或未来人那类的生物，
并拥有未知的能力，
后来他跟坏人战斗，
而我只要设法让自己被卷进那场战争就好了。
主要战斗的人是他，
而我则是追随他的小跟班。
啊啊，
实在太棒了，
我真是聪明啊!
要不然就是这样。
某天，
我那不可思议的力量突然觉醒，
就像隔空取物或精神念力之类的。
而且地球上其实还有很多拥有超能力的人类存在，
自然也会有一个组织专门收容这些人。
不久之后，
善良的组织便派人来迎接我。
而我也成为组织的一员共同对抗企图征服世界的邪恶超能力者。
不过，
现实却是意外地残酷。
现实的生活中，
并没有半个转学生转来我班上，
我也没看过UFO…
就算去了地方上常出现幽灵或妖怪的灵异地点也连个鬼影都没有…
花了两小时盯着桌上的铅笔，
它却连一微米都没移动…
上课时死盯着前座同学的头，
却怎么样也无法读出他在想什么。
我就这样边惊叹世界物理法则经常出现的现象，
边不停自嘲，
不知从何时起，
我就开始不看电视上的UFO特别节目或灵异节目了。
因为不可能会有那种东西……
不过后来我也成长到仅对那方面的事情存有一丝留恋的程度。
国中毕业之后，
我便从那孩提时代的梦想毕业，
逐渐习惯这个世界的平凡。
而让我还有一缕期待的一九九九年也没有发生什么事。
进入二十一世纪后，
人类依旧无法迈出月球到其他星球去。
看这情况，
在我还活着的时候，
想从地球当天往返阿尔法人马座似乎是不太可能的。
我脑海中时而幻想着这些事，
终于也没啥感慨地成为高中生 —— 直到遇到了凉宫春日。
之后，
我就这么糊里糊涂地进入学区内的县立高中就读。
起初我还很后悔，
因为这座学校位于很高的山上，
就算是春天也要挥汗如雨地爬上直窜山顶的坡道，
想轻松健行的那份悠闲早已消失无踪。
一想到今后三年每天一大早都得这样爬山，
我的心情就阴郁无比。
或许是早上差点睡过头的关系，
走路的速度自然加快许多，
虽然也曾想过以后干脆早十分钟起床，
慢慢走去上学就不会这么累，
不过一想到临起床前的那十分钟睡眠是多么的宝贵，
我随即放弃了这个念头。
所以，
我相信未来的三年还是得持续这个晨间的运动。
一想到这里，
心情就更沉重了。

*/