Cod sursa(job #2162672)

Utilizator albucristianAlbu Cristian-Gabriel albucristian Data 12 martie 2018 12:44:11
Problema Flux maxim Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.2 kb
#include <fstream>
#include <vector>
using namespace std;
int ma[1002][1002];
int gri[1002],gre[1002],viz[1002],tata[1002];
int n,m,a,b,c,q,w,sol,mi,cop,ok;
void dfs(int nod)
{
    viz[nod]=1;
    for(int i=1;i<=n;i++)
    {
        if(!ok)
            return;
        if(ma[nod][i]>0&&!viz[i])
        {
            if((ma[nod][i]<mi||mi==0))
            {
                mi=ma[nod][i];
            }
            tata[i]=nod;
            if(i==w)
                ok=0;
            dfs(i);
        }
    }
}
int main()
{
    ifstream in("maxflow.in");
    ofstream out("maxflow.out");
    in>>n>>m;
    for(int i=1;i<=m;i++)
    {
        in>>a>>b>>c;
        ma[a][b]=c;
        gri[b]++;
        gre[a]++;
    }
    for(int i=1;i<=n;i++)
    {
        if(gri[i]==0)
            q=i;
        if(gre[i]==0)
            w=i;
    }
    mi=sol=0;
    ok=1;
    dfs(q);
    while(viz[w])
    {
        sol+=mi;
        cop=w;
        while(tata[cop])
        {
            ma[tata[cop]][cop]-=mi;
            cop=tata[cop];
        }
        for(int i=1;i<=n;i++)
            viz[i]=0;
        ok=1;
        mi=0;
        dfs(q);
    }
    out<<sol;
    return 0;
}