2234. 多源汇最大流（最大流，超级源汇点）

活动 - AcWing

给定一个包含 n 个点 m 条边的有向图，并给定每条边的容量，边的容量非负。

其中有 Sc 个源点，Tc 个汇点。

图中可能存在重边和自环。

求整个网络的最大流。

输入格式

第一行包含四个整数 n,m,Sc,Tc。

第二行包含 Sc 个整数，表示所有源点的编号。

第三行包含 Tc 个整数，表示所有汇点的编号。

接下来 m 行，每行三个整数 u,v,c，表示从点 u 到点 v 存在一条有向边，容量为 c。

点的编号从 1 到 n。

输出格式

输出一个整数表示整个网络的最大流。

数据范围

2≤n≤10000
1≤m≤105
0≤c≤10000
保证源点集合和汇点集合没有交集。

输入样例：

4 5 2 2
2 4
1 3
4 2 30
4 3 20
2 3 20
2 1 30
1 3 40

输出样例：

70

解析： 

建立超级源点 S ，S 与所有源点连一条容量为正无穷的边；建立超级汇点 T ，T 与所有汇点连一条容量为正无穷的边。

正确性的证明也很简单，只需要证明最大流对应原图的最大流，原图的最大流对应新图的最大流即可。

#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<math.h>
#include<map>
#include<sstream>
#include<deque>
#include<unordered_map>
#include<unordered_set>
#include<bitset>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 1e4 + 10, M = (1e5 + N) * 2, INF = 0x3f3f3f3f;
int n, m, S, T;
int h[N], e[M], f[M], ne[M], idx;
int q[N], d[N], cur[N];

void add(int a, int b, int c) {
	e[idx] = b, f[idx] = c, ne[idx] = h[a], h[a] = idx++;
	e[idx] = a, f[idx] = 0, ne[idx] = h[b], h[b] = idx++;
}

bool bfs() {
	int hh = 0, tt = 0;
	memset(d, -1, sizeof d);
	q[0] = S, d[S] = 0, cur[S] = h[S];
	while (hh <= tt) {
		int t = q[hh++];
		for (int i = h[t]; i != -1; i = ne[i]) {
			int j = e[i];
			if (d[j] == -1 && f[i]) {
				d[j] = d[t] + 1;
				cur[j] = h[j];
				if (j == T)return 1;
				q[++tt] = j;
			}
		}
	}
	return 0;
}

int find(int u, int limit) {
	if (u == T)return limit;
	int flow = 0;
	for (int i = cur[u]; i != -1 && flow < limit; i = ne[i]) {
		int j = e[i];
		cur[u] = i;
		if (d[j] == d[u] + 1 && f[i]) {
			int t = find(j, min(f[i], limit - flow));
			if (!t)d[j] = -1;
			f[i] -= t, f[i ^ 1] += t, flow += t;
		}
	}
	return flow;
}

int dinic() {
	int ret = 0, flow;
	while (bfs())while (flow = find(S, INF))ret += flow;
	return ret;
}

int main() {
	int sc, tc;
	cin >> n >> m >> sc >> tc;
	memset(h, -1, sizeof h);
	S = 0, T = n + 1;
	for (int i = 1,a; i <= sc; i++) {
		scanf("%d", &a);
		add(S, a, INF);
	}
	for (int i = 1, a; i <= tc; i++) {
		scanf("%d", &a);
		add(a, T, INF);
	}
	for (int i = 1,a,b,c; i <= m; i++) {
		scanf("%d%d%d", &a, &b, &c);
		add(a, b, c);
	}
	printf("%d\n", dinic());
	return 0;
}