Cod sursa(job #1649808)

Utilizator alexmisto342Turdean Alexandru alexmisto342 Data 11 martie 2016 15:11:21
Problema Flux maxim Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.48 kb
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#define x first
#define y second
#define mp make_pair
using namespace std;
ifstream fin("maxflow.in");
ofstream fout("maxflow.out");
int i,j,n,m,p[1005],a,b,c[1001][1001],nod,ok,mini;
long long int flow;
vector <int > v[1005];
bool bfs()
{
   queue<int> q;
   q.push(1);
   p[1] = -1;
   int i;
   while(q.size())
   {
       nod = q.front(),q.pop();
       for(i = 0; i < v[nod].size(); i++)
       {
           int nxt = v[nod][i];
           if(p[nxt] == 0 && c[nod][nxt])
           {
               p[nxt] = nod;
               if(nxt != n)
                    q.push(nxt);
           }
       }
   }
   return p[n]>0;
}
int main()
{
    fin >> n >> m;
    for(i = 1; i <= m; i++)
    {
        fin >> a >> b >> j;
        v[a].push_back(b);
        v[b].push_back(a);
        c[a][b] = j;
    }
    while(bfs())
    {
        for(j = 0; j < v[n].size(); j++)
        {
            a = v[n][j];
            if(p[a] && v[a][j])
            {
                mini = v[a][j];
                for(i = a; i>1; i = p[i])
                    mini = min(mini, c[p[i]][i]);
                for(i = a; i>1; i = p[i])
                {
                     c[p[i]][i]-=mini;
                     c[i][p[i]]+=mini;
                }
                flow += mini;
                c[a][n] -= mini;
            }
        }
        memset(p,0,sizeof(p));
    }
    fout<<flow;
return 0;
}