POJ3281 Dining 最大流

题意

• n头牛，f种食物，d种饮料
• 每头牛会有喜欢的食物和饮料
• 每种食物和饮料只能供给一头牛
• 牛可以喜欢多种食物和饮料
• 问最多可以让多少头牛同时享受到喜欢的食物和饮料

思路

• 我们把食物，牛和饮料作为节点，把“喜欢”作为无向边
• 这样就可以连出一个三阶段的图
• 如果我们定向这些边，然后把食物和虚拟源点相连，饮料和虚拟汇点相连，边的容量为1，可以通过求解最大流来求解吗？
• 这样显然还有问题，因为牛的节点其实只能用一次，否则就会把一只牛重复计数了，所以可以认为牛节点也是有流量限制的
• 这样我们可以把牛节点拆成两个点，然后两点间有一条容量为1的有向边，就可以解决问题了。

实现

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <cmath>
#include <queue>
#include <iostream>
using namespace std;


const int INF = 0x3f3f3f3f;

//Dinic最大流, 节点编号从0开始 
struct MaxFlow{
    const static int MAX_V = 1005;
    int V;
    //终点、容量、反向边 
    struct edge{
        int to, cap, rev;
    };
    vector<edge> G[MAX_V];
    int level[MAX_V];//顶点到源点的距离标号 
    int iter[MAX_V];// 当前弧，在其之前的边已经没有用了

    void add_edge(int from, int to, int cap){
        G[from].push_back((edge){to, cap, (int)G[to].size()});
        G[to].push_back((edge){from, 0, (int)G[from].size()-1});
    } 

    // 通过BFS计算从源点出发的距离标号 
    void bfs(int s){
        fill(level, level + V, -1);
        queue<int> que;
        level[s] = 0;
        que.push(s);
        while (!que.empty()){
            int v = que.front();
            que.pop();
            for (int i=0; i< G[v].size(); i++){
                edge& e = G[v][i];
                if (e.cap > 0 && level[e.to] < 0){
                    level[e.to] = level[v] + 1;
                    que.push(e.to);
                }
            }
        }
    }

    //通过DFS寻找增广路 
    int dfs(int v, int t, int f){
        if (v == t)
            return f;
        for (int &i = iter[v]; i < G[v].size(); i++){
            edge& e = G[v][i];
            if (e.cap > 0 && level[v] < level[e.to]){
                int d = dfs(e.to, t, min(f, e.cap));
                if (d > 0){
                    e.cap -= d;
                    G[e.to][e.rev].cap += d;
                    return d;
                }
            }
        }
        return 0;
    }

    //求解从s到t的最大流 
    int max_flow(int s, int t){
        int flow = 0;
        for (;;){
            bfs(s);
            if(level[t] < 0)
                return flow;
            fill(iter, iter + V, 0);
            int f;
            while ((f = dfs(s, t, INF)) > 0){
                flow += f;
            }
        }
    }

    void init(int n = 0){
        for (int i = 0; i < V; i++){
            G[i].clear();
        }
        V = n;
    }
}mf;


int n,f,d;
int main(){
    ios::sync_with_stdio(false);
    cin>>n>>f>>d;
    mf.V = n * 2 + f + d + 2;
    for (int i=n*2+1;i <= n*2+f; i++){
        mf.add_edge(0, i, 1);
    }
    for (int i=n*2+f+1;i < mf.V-1; i++){
        mf.add_edge(i, mf.V-1, 1);
    }
    for (int i=1;i<=n;i++){
        int fi,di;
        mf.add_edge(i, i+n, 1);
        cin>>fi>>di;
        for (int j=0;j<fi;j++){
            int tmp;
            cin>>tmp;
            mf.add_edge(tmp+n*2, i, 1);
        }
        for (int j=0;j<di;j++){
            int tmp;
            cin>>tmp;
            mf.add_edge(i+n, tmp+n*2+f, 1);
        }
    }
    cout << mf.max_flow(0, mf.V-1) << endl;
    return 0;
}