【ybt金牌导航3-2-3】物品配对

物品配对

题目链接：ybt金牌导航3-2-3

题目大意

有三种物品，第一种物品可以与一些第二种、第三种的物品组合，在第二种可以选的里面选一个，在第三种可以选的里面选一个，就可以与这个第一种的组合。
然后每个第二种，第三种物品只能和一个第一种物品组合。
问你最多有多少个组合。

思路

看到组合想到网络流。

我们把第一二三中分别看成 $A,B,C$。
那由于第一种和第二第三种都有联系，那我们不能 $S\to A\to B\to C\to T$，因为这样不能让 $B$ 每个最多被选一次，也不能让 $A$ 要么都不要要么要两个。
那我们考虑把 $A$ 放到中间，变成 $S\to B\to A\to C\to T$。
但你会发现它就变成不能放 $A$ 每个最多被选一次，那我们考虑拆点，变成 $S\to B\to A_1\to A_2\to C\to T$，$A_1\to A_2$ 的边的流量是 $1$，那 $A$ 最多就只会被选一次了。

代码

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#define INF 0x3f3f3f3f3f3f3f3f 

using namespace std;

struct node {
	int x, to, nxt, op;
}e[8000001];
int n, f, d, S, T, le[1001], lee[1001];
int fn[101][101], dn[101][101];
int dis[1001], KK, tot_num;
queue <int> q;

void add(int x, int y, int z) {
	e[++KK] = (node){z, y, le[x], KK + 1}; le[x] = KK;
	e[++KK] = (node){0, x, le[y], KK - 1}; le[y] = KK;
}

bool bfs() {
	for (int i = 1; i <= tot_num; i++) {
		dis[i] = -1;
		lee[i] = le[i];
	}
	while (!q.empty()) q.pop();
	
	q.push(S);
	dis[S] = 0;
	while (!q.empty()) {
		int now = q.front();
		q.pop();
		
		for (int i = le[now]; i; i = e[i].nxt)
			if (e[i].x > 0 && dis[e[i].to] == -1) {
				dis[e[i].to] = dis[now] + 1;
				if (e[i].to == T) return 1;
				q.push(e[i].to); 
			}
	}
	
	return 0;
}

int dfs(int now, int sum) {
	if (now == T) return sum;
	
	int go = 0;
	for (int &i = lee[now]; i; i = e[i].nxt)
		if (e[i].x > 0 && dis[e[i].to] == dis[now] + 1) {
			int this_go = dfs(e[i].to, min(sum - go, e[i].x));
			if (this_go) {
				e[i].x -= this_go;
				e[e[i].op].x += this_go;
				go += this_go;
				if (go == sum) return go;
			}
		}
	
	if (go < sum) dis[now] = -1;
	return go;
}

int dinic() {
	int re = 0;
	while (bfs())
		re += dfs(S, INF);
	return re;
}

int main() {
	scanf("%d %d %d", &n, &f, &d);
	
	S = n + n + f + d + 1;
	T = n + n + f + d + 2;
	tot_num = T;
	
	for (int i = 1; i <= f; i++)//起点->B
		add(S, i, 1);
	for (int i = 1; i <= d; i++)//C->终点
		add(f + n + n + i, T, 1);
	for (int i = 1; i <= n; i++)//A1->A2
		add(f + i, f + n + i, 1);
	
	for (int i = 1; i <= n; i++) {
		scanf("%d %d", &fn[i][0], &dn[i][0]);
		
		for (int j = 1; j <= fn[i][0]; j++) {
			scanf("%d", &fn[i][j]);
			add(fn[i][j], f + i, 1);//B->A1
		}
		for (int j = 1; j <= dn[i][0]; j++) {
			scanf("%d", &dn[i][j]);
			add(f + n + i, f + n + n + dn[i][j], 1);//A2->C
		}
	}
	
	printf("%d", dinic());//跑最大流
	
	return 0;
}