Cod sursa(job #1535093)

Utilizator dobrebogdanDobre Bogdan Mihai dobrebogdan Data 24 noiembrie 2015 12:34:02
Problema Flux maxim de cost minim Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 2.1 kb
#include<cstdio>
#include<deque>
#include<algorithm>
#include<vector>
const int nmax=350;
const int mmax=12500;
using namespace std;
int n,m,s,t;
int flux[nmax+5][nmax+5];
int ca[nmax+5][nmax+5],co[nmax+5][nmax+5],ta[nmax+5],cv[nmax+5];
deque<int> que;
bool be[nmax+5],inq[nmax+5];
vector<int > ve[nmax+5];
bool belman()
{
    int x,i;
    for(i=1; i<=n; i++)
        cv[i]=2000000000;
    cv[s]=0;
    que.push_back(s);
    inq[s]=1;
    while(!que.empty())
    {
        x=que.front();
        for(i=ve[x].size()-1; i>=0; i--)
            if(ca[x][ve[x][i]]-flux[x][ve[x][i]]>0 && cv[x]+co[x][ve[x][i]]<cv[ve[x][i]])
            {
                cv[ve[x][i]]=cv[x]+co[x][ve[x][i]];
                ta[ve[x][i]]=x;
                if(inq[ve[x][i]]==0)
                {
                    inq[ve[x][i]]=1;
                    que.push_back(ve[x][i]);
                }
            }
        inq[x]=0;
        que.pop_front();
    }
    if(ta[t]!=-1)
        return 1;
    else
        return 0;
}
void da()
{
    int i,cmin,z;
    for(i=1; i<=n; i++)
        ta[i]=-1;
    cmin=-1;
    // i=0;
    while(belman())
    {
        z=t;
        while(z!=s)
        {
            if(ca[ta[z]][z]-flux[ta[z]][z]<cmin || cmin==-1)
                cmin=ca[ta[z]][z]-flux[ta[z]][z];
            z=ta[z];
        }
        z=t;
        while(z!=s)
        {
            flux[ta[z]][z]+=cmin;
            flux[z][ta[z]]-=cmin;
            z=ta[z];
        }
        for(i=1; i<=n; i++)
            ta[i]=-1;
    }
}
int main()
{
    freopen("fmcm.in","r",stdin);
    freopen("fmcm.out","w",stdout);
    int i,j,x,y,cmin;
    scanf("%d%d%d%d",&n,&m,&s,&t);
    for(i=1; i<=m; i++)
    {
        scanf("%d%d",&x,&y);
        scanf("%d%d",&ca[x][y],&co[x][y]);
        co[y][x]=-co[x][y];
        ve[x].push_back(y);
        ve[y].push_back(x);
    }
    da();
    cmin=0;
    for(i=1; i<=n; i++)
        for(j=1; j<=n; j++)
            if(flux[i][j]>0)
            {
                cmin=cmin+flux[i][j]*co[i][j];
            }
    printf("%d\n",cmin);
    return 0;
}