POJ 3281 （网络流）

题意：

有N头牛，F种食物，D种饮料，每种食物和饮料只能提供给一头牛。每头牛有自
己喜欢的饮料和食物。问最多有多少头牛能吃到自己喜欢的饮料和食物。

思路：

一般的网络流是求出最大流量，并不能解决有两个选择的题。但是此题可以再
增加一排牛建图，这样可以用网络最大流解决。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>

using namespace std;

const int MAXN = 10005;
const int INF = 0x3f3f3f3f;

struct edge
{
    int to,cap,rev;
};

int N,F,D;
vector<edge>G[MAXN];
int level[MAXN];
int iter[MAXN];

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

void bfs(int s)
{
    memset(level,-1,sizeof(level));
    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);
            }
        }
    }
}

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 false;
}

int max_flow(int v,int t)
{
    int flow = 0;
    int f;
    for( ;; ) {
        bfs(v);
        if(level[t] < 0) return flow;
        memset(iter,0,sizeof(iter));
        while((f = dfs(v,t,INF)) > 0) {
            flow += f;
        }
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    scanf("%d%d%d",&N,&F,&D);
    for(int i = 0;i <= N+F+D+N+100; i++)
        G[i].clear();
    for(int i = 1;i <= F; i++) {
        add_edge(0,i,1);
    }
    for(int i = 1;i <= D; i++) {
        add_edge(i+F+N*2,F+N*2+D+1,1);
    }
    for(int i = 1;i <= N; i++) {
        add_edge(i+F,F+i+N,1);
    }
    for(int i = 1;i <= N; i++) {
        int f,d;
        scanf("%d%d",&f,&d);
        for(int j = 1;j <= f; j++){
            int temp;
            scanf("%d",&temp);
            add_edge(temp,i+F,1);
        }
        for(int j = 1;j <= d; j++) {
            int temp;
            scanf("%d",&temp);
            add_edge(F+i+N,temp+F+N*2,1);
        }
    }
    printf("%d\n",max_flow(0,F+D+2*N+1));
    return 0;
}