POJ【3281】Dining（网络流）

题目链接：http://poj.org/problem?id=3281
分析
这道题的意思是有N头牛，F种食物和D种饮料，每头牛有自己喜欢的食物和饮料且只能选择一种食物和饮料，所有牛不能选择相同的食物和饮料，问最多有多少头牛能得到满足。这道题类似二分匹配，但又不全是，个人理解有点像将二分匹配转化为特殊的网络流的过程。以下是图的模型。

增加一个源点s和一个汇点t，将牛拆分成两份，源点指向食物，食物指向牛，牛指向拆分后的自己，牛再指向饮料，饮料指向汇点，所有的边的权值设置为1。个人理解边的权值为1，所以最终一定是一头牛对应一种食物和饮料。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int INF = 0x3f3f3f3f;
const int N = 1000 + 10;
int n,f,d,m;
int a[N][N],dist[N];
bool bfs(int s,int t)
{
    memset(dist,-1,sizeof(dist));
    queue<int> q;
    q.push(s);
    dist[s] = 0;
    while(!q.empty()) {
        int u = q.front();q.pop();
        for(int v = 1; v <= m; v++) {
            if(dist[v] == -1 && a[u][v]) {
                dist[v] = dist[u] + 1;//记录点v的层次
                q.push(v);
            }
        }
    }
    return dist[t] != -1;
}
int dfs(int s,int t,int res)//res表示最小的残量
{
    if(s == t || res == 0) return res;
    int ans = 0,f;
    for(int v = 1; v <= m; v++) {
        if(dist[v] == dist[s] + 1 && (f = dfs(v,t,min(a[s][v],res))) > 0) {
            a[s][v] -= f;
            a[v][s] += f;
            ans += f;
            res -= f;
            if(res == 0) break;
        }
    }
    return ans;
}
int Dinic(int s,int t)
{
    int ans = 0;
    while(bfs(s,t)) ans += dfs(s,t,INF);
    return ans;
}
int main()
{
    //建图
    scanf("%d%d%d",&n,&f,&d);
    int x,y;
    //s为源点
    int s = 2 * n + f + d + 1;
    //t为汇点
    int t = s + 1;
    m = t;
    for(int i = 1; i <= f; i++)
        a[s][i + 2 * n] = 1;//源点指向食物
    for(int i = 1; i <= d; i++)
        a[i + 2 * n + f][t] = 1;//饮料指向汇点
    for(int i = 1,j,k; i <= n; i++) {
        a[i][i + n] = 1;//牛指向牛
        scanf("%d%d",&x,&y);
        for(int k = 0; k < x; k++) {
            scanf("%d",&j);
            a[j + 2 * n][i] = 1;//食物指向牛
        }
        for(int k = 0; k < y; k++) {
            scanf("%d",&j);
            a[i + n][j + 2 * n + f] = 1;//牛指向饮料
        }
    }
    int ans = Dinic(s,t);
    printf("%d\n",ans);
    return 0;
}