Cod sursa(job #2453997)

Utilizator victoreVictor Popa victore Data 7 septembrie 2019 01:12:27
Problema Flux maxim Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.78 kb
//check check check
//#include<iostream>
#include<vector>
#include<algorithm>
#include<fstream>
#include<queue>
#include<cstring>
#include<map>
#include<iomanip>

#define ll long long
#define pb(x) push_back(x)

using namespace std;

typedef pair<int,int> ii;

const int NMAX = 105;
const int INF = 1e9;

ifstream cin("maxflow.in");
ofstream cout("maxflow.out");

vector<int>g[NMAX];

int capacity[NMAX][NMAX],flow[NMAX][NMAX];
int N,M,tata[NMAX];


int BFS(int source , int sink)
{
    int i;
    queue<ii>q;
    for(i = 1 ; i <= N ; ++i)
        tata[i] = 0;
    tata[source] = -1;
    q.push(ii(source , INF));
    int node,curr_flow;
    while(!q.empty())
    {
        node = q.front().first;
        curr_flow = q.front().second;
        q.pop();
        for(auto gnode : g[node])
        {
            if(tata[gnode] == 0 && capacity[node][gnode] > flow[node][gnode])
            {
                tata[gnode] = node;
                if(gnode == sink)
                {
                    return min(curr_flow , capacity[node][gnode] - flow[node][gnode]);
                }
                q.push(ii(gnode , min(curr_flow , capacity[node][gnode] - flow[node][gnode])));
            }
        }
    }

    return 0;
}

int main()
{
    int i,j,x,c,y;
    cin>>N>>M;

    for(i = 1 ; i <= M ; ++i)
    {
        cin>>x>>y>>c;
        g[x].pb(y);
        g[y].pb(x);
        capacity[x][y] = c;
    }

    int new_flow,source = 1 , sink = N , maxflow = 0;
    while(new_flow = BFS(source , sink))
    {
        for(i = sink ; i != source ; i = tata[i])
        {
            flow[tata[i]][i] += new_flow;
            flow[i][tata[i]] -= new_flow;
        }
        maxflow += new_flow;
    }

    cout<<maxflow;

    return 0;
}