Cod sursa(job #2471094)

Utilizator Cristian1997Vintur Cristian Cristian1997 Data 10 octombrie 2019 11:31:39
Problema Flux maxim Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.17 kb
#include <bits/stdc++.h>
using namespace std;
using uint = unsigned int;
using ll = long long;
using ull = unsigned long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
#define dbg(x) cerr<<#x": "<<(x)<<endl
#define dbg_p(x) cerr<<#x": "<<(x).first<<' '<<(x).second<<endl
#define dbg_v(x, n) {cerr<<#x"[]: ";for(long long _=0;_<n;++_)cerr<<(x)[_]<<' ';cerr<<endl;}
#define all(v) v.begin(), v.end()
#define fi first
#define se second

template<typename T1, typename T2>
ostream& operator <<(ostream &out, const pair<T1, T2> &item) {
	out << '(' << item.first << ", " << item.second << ')';
	return out;
}

template <typename T>
ostream& operator <<(ostream &out, const vector<T>& v) {
	for(const auto &item : v) out << item << ' ';
	return out;
}

const int N = 1010;
const int INF = 0x3f3f3f3f;

struct Edge {
	int to, flow;
};

int source, sink;
int d[N];
bool vis[N];
vector<int> adj[N];
vector<Edge> edges;

void addEdge(int a, int b, int cap) {
	adj[a].push_back(edges.size());
	edges.push_back({b, cap});
	adj[b].push_back(edges.size());
	edges.push_back({a, 0});
}

bool bfs(int source) {
	memset(d, INF, sizeof d);

	queue<int> q;
	for(q.push(source), d[source] = 0; !q.empty(); q.pop()) {
		for(auto id : adj[q.front()])
			if(edges[id].flow && d[edges[id].to] == INF) {
				d[edges[id].to] = 1 + d[q.front()];
				q.push(edges[id].to);
			}
	}

	return d[sink] != INF;
}

int dfs(int v, int flow) {
	if(v == sink || !flow) return flow;
	if(vis[v]) return 0;
	vis[v] = true;

	for(auto id : adj[v]) {
		int u = edges[id].to;
		if(d[u] == 1 + d[v]) {
			int pushed = dfs(u, min(flow, edges[id].flow));
			if(pushed) {
				edges[id].flow -= pushed;
				edges[id ^ 1].flow += pushed;
				return pushed;
			}
		}
	}

	return 0;
}

int dinic(int source) {
	int flow = 0, pushed;
	while(bfs(source)) {
		memset(vis, 0, sizeof vis);
		while(pushed = dfs(source, INF)) flow += pushed;
	}

	return flow;
}

int main()
{
	freopen("maxflow.in", "r", stdin);
	freopen("maxflow.out", "w", stdout);
	ios_base::sync_with_stdio(false);
	
	int n, m, a, b, c;

	cin >> n >> m;
	source = 1;
	sink = n;

	for(int i = 1; i <= m; ++i) {
		cin >> a >> b >> c;
		addEdge(a, b, c);
	}

	cout << dinic(source) << '\n';
	return 0;
}