Dining （POJ No.3281) 网络流经典题

• 题意：有N头牛，F种食物，D种饮料，每头牛都有自己喜欢的食物和饮料，每种饮料和食物只能分配给一头牛。问：最多有多少头牛能同时得到自己喜欢的食物和饮料。
• 思路：挑战上面的例题，真的很经典，虽然知道是网络流不看书上给的思路怎么也想不到这道题应该这样建图。最开始我的想法是，给牛和喜欢的食物建立一条容量为1的边，同时和喜欢的饮料建立一条容量为1的边，s（源点）和每一种食物建立一条容量为1的边，每一种饮料和t（汇点）建立一条容量为1的边，然后求解s到t的最大流就是答案。但是这样有一个问题，就是一头牛可能匹配多对食物和饮料，而题目的意思应该是每头牛只能得到一种食物和一种饮料。所以这道题的关键是如果限制每头牛只能得到一种食物和一种饮料呢？《挑战》上面给出了一个极妙的解决办法，把每头牛分为两个点，然后一个点和它喜欢的食物建立边，另外一个点和它喜欢的饮料建立一条边，再在同一头牛的两个点之间建立一条容量为2的边，这样就保证了每头牛一定只能得到一种食物和一种饮料。 如果这道题变为每头牛最多可以同时得到两种食物和两种饮料，并且牛得到一种食物同时必定会得到一种饮料，问所有牛得到的自己喜欢的食物一共有多少种。 这样只需把同一个牛的两个点之间的边的容量设置为2就好了。
• 代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#include <queue>
#include <algorithm>
#include <map>
#include <set>


using namespace std;


const int INF = 0x3f3f3f3f;
const int MAX_V = 1000;

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

vector<edge> G[MAX_V];

bool used[MAX_V];

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

int dfs(int v, int t, int f) {
    if(v == t) return f;
    used[v] = true;
    for(int i = 0; i < G[v].size(); i++) {
        edge &e = G[v][i];
        if(!used[e.to] && e.cap > 0) {
            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;
}

int max_flow(int s, int t) {
    int flow = 0;
    for(;;) {
        memset(used, 0, sizeof(used));
        int f = dfs(s, t, INF);
        if(f > 0) flow += f;
        else {
            return flow;
        }
    }
}

int main() {
    int N, F, D;
    while(~scanf("%d%d%d", &N, &F, &D)) {
        //initialize
        int V = 2*N + F + D + 2;
        int s = 2*N + F + D, t = s + 1;
        for(int v = 0; v < V; v++) {
            G[v].clear();
        }

        for(int i = 0; i < F; i++) {
            add_edge(s, 2*N + i, 1);
        }
        for(int i = 0; i < N; i++) {
            add_edge(i, N + i, 1);
        }
        for(int i = 0; i < D; i++) {
            add_edge(2*N + F + i, t, 1);
        }

        //input
        for(int i = 0; i < N; i++) {
            int cd = 0, cf = 0;
            scanf("%d%d", &cf, &cd);
            for(int j = 0; j < cf; j++) {
                int fi;
                scanf("%d", &fi);
                add_edge(2*N + fi - 1, i, 1);
            }

            for(int j = 0; j < cd; j++) {
                int di;
                scanf("%d", &di);
                add_edge(N + i, 2*N + F + di - 1, 1);
            }
        }
        int ans = max_flow(s, t);
        printf("%d\n", ans);
    }
    return 0;
}

• 一个奇怪的问题： 这道题中我使用了Ford-Fulkerson和Dinic两种网络流的模版提交，在Poj上For-Fulkerson（也就是上面这个）能ac，而Dinic却wa掉了，而在soj上则两个都能ac，求大神告知这是为什么。这里附上我写的Dinic版本。

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

using namespace std;

const int INF = 0x3f3f3f3f;
const int MAX_V = 1000;

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, G[to].size()});
    G[to].push_back((edge){from, cap, 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);
            }
        }
    }
}

//通过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(e.cap, f));
            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;
        memset(iter, 0, sizeof(iter));
        int f;
        while((f = dfs(s, t, INF)) > 0) {
            flow += f;
        }
    }
}


int main() {
    int N, F, D;
    while(~scanf("%d%d%d", &N, &F, &D)) {
        //initialize
        int V = 2*N + F + D + 2;
        int s = 2*N + F + D, t = s + 1;
        for(int v = 0; v < V; v++) {
            G[v].clear();
        }

        for(int i = 0; i < F; i++) {
            add_edge(s, 2*N + i, 1);
        }
        for(int i = 0; i < N; i++) {
            add_edge(i, N + i, 1);
        }
        for(int i = 0; i < D; i++) {
            add_edge(2*N + F + i, t, 1);
        }

        //input
        for(int i = 0; i < N; i++) {
            int cd = 0, cf = 0;
            scanf("%d%d", &cf, &cd);
            for(int j = 0; j < cf; j++) {
                int fi;
                scanf("%d", &fi);
                add_edge(2*N + fi - 1, i, 1);
            }

            for(int j = 0; j < cd; j++) {
                int di;
                scanf("%d", &di);
                add_edge(N + i, 2*N + F + di - 1, 1);
            }
        }
        int ans = max_flow(s, t);
        printf("%d\n", ans);
    }
    return 0;
}