poj 3281 Dining(拆點+最大流)
阿新 • • 發佈:2018-11-06
POJ 3281 : 題目連結
題目大意是N頭牛,準備了F種食物,D種飲料,每一頭牛會喜歡若干種食物和飲料,但它只能選擇一種食物和一種飲料,且每種食物和飲料都只被一頭牛選擇,問怎樣分配能使得食物和飲料都能得到的牛的數量最多,求牛的個數
建模
1,因為每頭牛都必須要選擇一種飲料一種食物 ,將食物和飲料放在兩邊 ,分別於源點匯點相連,把牛放在中間 ,與其有關聯的食物飲料相連。
2,每頭牛都只能選擇 一種飲料 一種食物 所以要用到拆點。
void add_map() { for (int i = 1; i <= f; i++) add_node(0, 2*n + i, 1);//將每個食物於源點相連 因為食物只能選一次 所以權值為1 for (int i = 1; i <= d; i++) add_node(2*n + f + i, 2*n + f + d + 1, 1);//每個飲料於匯點相連 選一次 for (int i = 1; i <= n; i++) add_node(i, n + i, 1);//將一個牛拆成兩個點 權值為1 保證每隻牛隻被選一次 }
for (int j = 1; j <= a; j++) {
scanf("%d", &tmp);
add_node(2 * n + tmp, i, inf);//將是食物 於牛相連 權值為無限大
}
for (int j = 1; j <= b; j++) {
scanf("%d", &tmp);
add_node(n + i, 2 * n + f + tmp, inf);//將飲料於牛相連
}
上模板
#include<iostream> #include<queue> #include<vector> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int maxn = 510; const int inf = 0x3f3f3f3f; int level[maxn], n, f, d; struct node { int to, cap, rev; node(int _to,int _cap,int _rev):to(_to),cap(_cap),rev(_rev){} }; vector<node>edge[maxn]; void add_node(int a, int b, int c) { //printf("from %d to %d is %d\n", a, b, c); edge[a].push_back(node(b, c, edge[b].size())); edge[b].push_back(node(a, 0, edge[a].size() - 1)); } int bfs(int s, int t) { memset(level, -1, sizeof(level)); queue<int>q; q.push(s); level[s] = 0; while (!q.empty()) { int tmp = q.front(); q.pop(); for (int i = 0; i < edge[tmp].size(); i++) { node &tmp1 = edge[tmp][i]; if (level[tmp1.to] == -1 && tmp1.cap > 0) { level[tmp1.to] = level[tmp] + 1; q.push(tmp1.to); } } } return level[t] != -1; } int dfs(int s, int t, int flow) { if (s == t) return flow; int ans = 0; for (int i = 0; i < edge[s].size(); i++) { node &tmp = edge[s][i]; if (flow > ans&&tmp.cap > 0 && level[tmp.to] == level[s] + 1) { int d = dfs(tmp.to, t, min(tmp.cap, flow - ans)); if (d > 0) { tmp.cap -= d; edge[tmp.to][tmp.rev].cap += d; ans += d; } } } if (ans == 0) level[s] = -1; return ans; } int dinic(int s, int t) { int ans = 0, tf; while (bfs(s, t)) while (tf = dfs(s, t, inf)) ans += tf; return ans; } void add_map() { for (int i = 1; i <= f; i++) add_node(0, 2*n + i, 1); for (int i = 1; i <= d; i++) add_node(2*n + f + i, 2*n + f + d + 1, 1); for (int i = 1; i <= n; i++) add_node(i, n + i, 1); } int main() { int a, b, tmp; scanf("%d%d%d", &n, &f, &d); add_map(); for (int i = 1; i <= n; i++) { scanf("%d%d", &a, &b); for (int j = 1; j <= a; j++) { scanf("%d", &tmp); add_node(2 * n + tmp, i, inf); } for (int j = 1; j <= b; j++) { scanf("%d", &tmp); add_node(n + i, 2 * n + f + tmp, inf); } } printf("%d\n", dinic(0, 2 * n + f + d + 1)); return 0; }