Cod sursa(job #1414052)

Utilizator rares96cheseliRares Cheseli rares96cheseli Data 2 aprilie 2015 12:13:07
Problema Flux maxim Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.67 kb
#include <fstream>
#include <queue>
#include <algorithm>
using namespace std;
ifstream f("maxflow.in");
ofstream g("maxflow.out");

int N, M, x, y, c, flux[1005][1005], cap[1005][1005], sol, dad[1005], start=1, dest;
vector < int > G[1005];
queue < int > Q;
bool inQ[1005];

bool bfs(int sursa)
{
    for (int i=1; i<=N; ++i)
        dad[i]=0;
    Q.push(sursa); dad[sursa]=sursa; inQ[sursa]=1;
    while (Q.size())
    {
        int nod=Q.front(); Q.pop(); inQ[nod]=0;
        if (nod==dest) continue;
        vector<int>::iterator it=G[nod].begin();
        for (; it!=G[nod].end(); ++it)
            if (!dad[*it] && cap[nod][*it]>flux[nod][*it])
            {
                dad[*it]=nod;
                if (!inQ[*it])
                    Q.push(*it), inQ[*it]=1;
            }
    }
    return (dad[dest]>0);
}

void go(int sursa)
{
    for (; bfs(sursa); )
    {
        vector<int>::iterator it=G[dest].begin();
        for (; it!=G[dest].end(); ++it)
            if (dad[dest]==*it)
            {
                int minim=(1<<30);
                for (int x=dest; x!=start; x=dad[x])
                    minim=min(minim, cap[dad[x]][x]-flux[dad[x]][x]);

                if (minim<(1<<30))
                {
                    for (int x=dest; x!=start; x=dad[x])
                        flux[dad[x]][x]+=minim, flux[x][dad[x]]-=minim;
                    sol+=minim;
                }
            }
    }
}

int main()
{
    f>>N>>M; dest=N;
    for (int i=1; i<=M; ++i)
    {
        f>>x>>y>>c;
        cap[x][y]=c;
        G[x].push_back(y);
        G[y].push_back(x);
    }

    go(1);
    g<<sol<<'\n';
    return 0;
}