Cod sursa(job #534461)

Utilizator R.A.RFMI Romila Remus Arthur R.A.R Data 15 februarie 2011 19:12:11
Problema Flux maxim Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.66 kb
#include <fstream>
#include <vector>
#include <queue>
#define nmax 1010
#define INF 1<<30

using namespace std;

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

int N,M,S,D,flux;
vector<int> V[nmax];
queue<int> Q;
int T[nmax];
int C[nmax][nmax];//capacitatea maxima pe muchii

inline int minim(int a,int b){if(a<b)return a;return b;}

int BF()
{
    while(!Q.empty())Q.pop();
    int i,x,y;
    for(i=0;i<=N;i++)T[i]=0;
    T[S] = -1;
    Q.push(S);
    while(!Q.empty())
    {
        x = Q.front();
        Q.pop();
        for(i=0;i<V[x].size();i++)
        {
            y =V[x][i];
            if(C[x][y]&&!T[y])//exista muchia
            {
                Q.push(y);
                T[y] = x;
                if(y==D)
                    return 1;
            }
        }
    }
    return 0;
}

int main()
{
    int x,y,c,muchie_minima,i;
    in>>N>>M;
    S=1,D=N;
    while(M--)
    {
        in>>x>>y>>c;
        C[x][y]+=c;
        V[x].push_back(y);
        V[y].push_back(x);
    }
    while(BF())
    {
        for(i=0;i<V[D].size();i++)
        {
            //am determinat un drum de la sursa la destinatie
            x = V[D][i];
            if (!C[x][D] || !T[x]) continue;
            T[D] = x;
            muchie_minima = INF;
            while(x!=S)
                muchie_minima = minim(muchie_minima,C[T[x]][x]),x=T[x];//determin costul minim
            if(!muchie_minima)continue;
            //actualizez costurile
            x=D;
            while(x!=S)
                C[T[x]][x]-=muchie_minima,x=T[x];
            flux+=muchie_minima;
        }
    }
    out<<flux;
    return 0;
}