Cod sursa(job #1409666)

Utilizator robertstrecheStreche Robert robertstreche Data 30 martie 2015 17:28:45
Problema Flux maxim de cost minim Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.69 kb
#include <cstdio>
#include <vector>
#include <cstring>
#include <queue>

#define NMAX 355
#define INF 0x3f3f3f3f3f

using namespace std;

bool ap[NMAX];

int boss[NMAX],best[NMAX];
int n,m,x,y,z,c,surs,dest,costmax;

int cap[NMAX][NMAX],cost[NMAX][NMAX],flow[NMAX][NMAX];

queue <int >q;
vector <int>v[NMAX];

inline bool bellman(int surs)
{
    memset(ap,0,sizeof(ap));
    for (int i=1;i<=n;i++)best[i]=INF,boss[i]=0;
    ap[surs]=1;
    q.push(surs);
    best[surs]=0;
    for (;!q.empty();q.pop())
    {
        int nod=q.front();
        ap[nod]=0;
        for (auto it:v[nod])
          if (best[it]>best[nod]+cost[nod][it] && cap[nod][it]>0)
          {
              boss[it]=nod;
              best[it]=best[nod]+cost[nod][it];
              if (!ap[it])q.push(it);
          }
    }
    return boss[dest]!=0;
}
int main()
{
   freopen("fmcm.in","r",stdin);
   freopen("fmcm.out","w",stdout);
   scanf("%d %d %d %d",&n,&m,&surs,&dest);
   for (int i=1;i<=m;i++)
   {
       scanf("%d %d %d %d",&x,&y,&z,&c);
       v[x].push_back(y);
       v[y].push_back(x);
       cost[x][y]=c;
       cost[y][x]=-c;
       cap[x][y]=z;
   }
   for (;bellman(surs);)
   {
       int maxflow=INF;
       for (int nod=dest;boss[nod];nod=boss[nod])
        maxflow=(maxflow>cap[boss[nod]][nod]?cap[boss[nod]][nod]:maxflow);
       for (int nod=dest;boss[nod];nod=boss[nod])
       {
           cap[boss[nod]][nod]-=maxflow;
           cap[nod][boss[nod]]+=maxflow;
           flow[boss[nod]][nod]+=maxflow;
           costmax+=cost[boss[nod]][nod]*maxflow;
       }
       //printf("%d\n",costmax);
   }
   printf("%d",costmax);

   fclose(stdin);
   fclose(stdout);
}