Cod sursa(job #3183214)

Utilizator Ilie_MityIlie Dumitru Ilie_Mity Data 11 decembrie 2023 10:26:25
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.98 kb
// Ilie Dumitru
#include<cstdio>
#include<vector>
#include<bitset>
#include<queue>
const int NMAX=1005, oo=(1LL<<31)-1;

struct edge
{
	int node, left, idxOther;
};

int N, M;
std::vector<edge> G[NMAX];
int ans;
int level[NMAX];
std::bitset<NMAX> saturated;

void addEdge(int u, int v, int w)
{
	G[u].push_back({v, w, (int)G[v].size()});
	G[v].push_back({u, 0, (int)G[u].size()-1});
}

void useEdge(edge& e, int flow)
{
	e.left-=flow;
	G[e.node][e.idxOther].left+=flow;
}

bool bfs(int source, int target)
{
	int i, node;
	std::queue<int> q;
	bool ok=0;

	q.push(source);
	for(i=0;i<N;++i)
		level[i]=NMAX;
	level[source]=0;

	do
	{
		node=q.front();
		q.pop();

		for(i=0;i<(int)G[node].size();++i)
		{
			if(level[node]+1<level[G[node][i].node] && G[node][i].left)
			{
				if(G[node][i].node!=target)
				{
					level[G[node][i].node]=1+level[node];
					q.push(G[node][i].node);
				}
				else
					ok=1;
			}
		}
	}while(!q.empty());

	return ok;
}

int dfs(int node, int target, int flow=oo)
{
	int i, x, nn, totalUsed=0;

	for(i=0;i<(int)G[node].size() && flow;++i)
	{
		nn=G[node][i].node;
		if(G[node][i].left)
		{
			if(nn==target)
			{
				x=std::min(flow, G[node][i].left);
				ans+=x;
				flow-=x;
				totalUsed+=x;
				useEdge(G[node][i], x);
			}
			else if(level[nn]>level[node])
			{
				x=std::min(flow, G[node][i].left);
				x=dfs(nn, target, x);
				totalUsed+=x;
				flow-=x;
				useEdge(G[node][i], x);
			}
		}
	}

	return totalUsed;
}

bool dinic(int source, int target)
{
	if(bfs(source, target))
	{
		saturated.reset();
		dfs(source, target);

		return 1;
	}

	return 0;
}

int main()
{
	FILE* f=fopen("maxflow.in", "r"), *g=fopen("maxflow.out", "w");
	int i, a, b, c;

	fscanf(f, "%d%d", &N, &M);
	for(i=0;i<M;++i)
	{
		fscanf(f, "%d%d%d", &a, &b, &c);
		addEdge(a-1, b-1, c);
	}

	while(dinic(0, N-1));

    fprintf(g, "%d\n", ans);

	fclose(f);
	fclose(g);
	return 0;
}