1. 程式人生 > >poj 3281 Dining(拆點+最大流)

poj 3281 Dining(拆點+最大流)

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