POJ 3281 （网络流）

来源：互联网发布：网络大电影方案编辑：程序博客网时间：2024/05/18 04:33

题意：

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

0 0