洛谷 P2071 座位安排（dinic网络流求二分图的最大匹配）

题目链接

洛谷 P2071 座位安排

思路

显然，这是一个求二分图的最大匹配数问题，我们可以使用dinic算法进行求解。

我们用$2n$个人和$n$排座位组成二分图。

对于第$i$个人想坐的排数$a$和$b$，我们从点$i$向$(n\times 2+a)$和$(n\times 2+b)$分别连一条容量为$1$的边。对于源点$S$，我们向每个人都连一条容量为$1$的边。对于每一排座位，我们向汇点$T$连一条容量为$2$的边。

对于我们建成的流网络，满足流量守恒和能量守恒。因此，求出的最大流即为最终的答案。

代码

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 6e3 + 5, M = 2e4 + 8e3 + 5;
const int inf = 0x3f3f3f3f3f3f3f3f;
int n, m, S, T;
int d[N];   // 点在分层图中的编号
int cur[N]; // 当前弧优化，存储从当前节点开始还需要遍历的点的编号
struct Edge
{
	int to, f, next; // 终点，容量，同起点的上一条边的编号
} edge[M];           // 边集
int head[N], idx;    // head[i],表示以i为起点的第一条边在边集数组的位置（编号）
void init(int n)          // 初始化
{
	for (int i = 0; i <= n; i++)
		head[i] = -1;
	idx = 0;
}
void add_edge(int u, int v, int f) // 加边，u起点，v终点，w容量
{
	edge[idx].to = v;         // 终点
	edge[idx].f = f;          // 权值
	edge[idx].next = head[u]; // 以u为起点上一条边的编号，也就是与这个边起点相同的上一条边的编号
	head[u] = idx++;          // 更新以u为起点上一条边的编号
}
bool bfs()
{
	queue<int> q;
	memset(d, -1, sizeof d);
	q.push(S);
	d[S] = 0;
	cur[S] = head[S];
	while (q.size())
	{
		int u = q.front();
		q.pop();
		for (int i = head[u]; i != -1; i = edge[i].next)
		{
			int j = edge[i].to;
			if (d[j] == -1 && edge[i].f)
			{
				d[j] = d[u] + 1;
				cur[j] = head[j];
				if (j == T)
					return true;
				q.push(j);
			}
		}
	}
	return false;
}
// 从起点S到当前点u允许流过的最大流量为limit
int find(int u, int limit)
{
	if (u == T)
		return limit;
	int flow = 0;
	for (int i = cur[u]; i != -1 && flow < limit; i = edge[i].next)
	{
		cur[u] = i; // 当前弧优化
		int j = edge[i].to;
		if (d[j] == d[u] + 1 && edge[i].f)
		{
			int op = find(j, min(edge[i].f, limit - flow));
			// 废点优化
			// op==0 说明不能找到从j到T的增广路
			// 因此下一次遍历到j时直接跳过即可, 将d[j]标记为-1即可
			if (!op)
				d[j] = -1;
			edge[i].f -= op;
			edge[i ^ 1].f += op;
			flow += op;
		}
	}
	return flow;
}
int dinic()
{
	int ans = 0, flow;
	while (bfs())
	{
		while (flow = find(S, inf))
			ans += flow;
	}
	return ans;
}
void solve()
{
	cin >> n;
	S = 0, T = n * 3 + 1;
	memset(head, -1, sizeof head);
	for (int i = 1, a, b; i <= n * 2; i++)
	{
		cin >> a >> b;
		add_edge(i, n * 2 + a, 1), add_edge(n * 2 + a, i, 0);
		add_edge(i, n * 2 + b, 1), add_edge(n * 2 + b, i, 0);
	}
	for (int i = 1; i <= n * 2; i++)
	{
		add_edge(S, i, 1), add_edge(i, S, 0);
	}
	for (int i = 1; i <= n; i++)
	{
		add_edge(n * 2 + i, T, 2), add_edge(T, n * 2 + i, 0);
	}
	int ans = dinic();
	cout << ans << endl;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int test = 1;
	// cin >> test;
	for (int i = 1; i <= test; i++)
	{
		solve();
	}
	return 0;
}