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