给定一个包含 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;
}