Pagini recente » Cod sursa (job #1694900) | Cod sursa (job #662644) | Cod sursa (job #1815397) | Cod sursa (job #1920340) | Cod sursa (job #1717068)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <bitset>
using namespace std;
const int inf=1000000000;
struct heap
{
int nod,c;
bool operator <(const heap &aux) const
{
return c>aux.c;
}
};
ifstream si("fmcm.in");
ofstream so("fmcm.out");
vector<int> v[355];
queue<int> q;
priority_queue<heap> h;
int cost[355][355],dist[355],dist1[355],cap[355][355],flow[355][355],d[355],tata[355],sursa,dest,n;
bitset<355>vaz;
void bf(int nod)
{
int i;
for(i=1;i<=n;i++)
dist[i]=inf;
dist[nod]=0;
q.push(nod);
vaz[nod]=1;
vector<int>::iterator it;
while(!q.empty())
{
int nod=q.front();
q.pop();
vaz[nod]=0;
for(it=v[nod].begin();it!=v[nod].end();it++)
if(cap[nod][*it]&&dist[*it]>dist[nod]+cost[nod][*it])
{
dist[*it]=dist[nod]+cost[nod][*it];
if(!vaz[*it]&&*it!=dest)
{
vaz[*it]=1;
q.push(*it);
}
}
}
}
int dijkstra()
{
int i;
for(i=1;i<=n;i++)
d[i]=dist1[i]=inf;
d[sursa]=dist1[sursa]=0;
h.push({sursa,0});
vector<int>::iterator it;
while(!h.empty())
{
int nod=h.top().nod;
int c=h.top().c;
h.pop();
if(d[nod]!=c)
continue;
for(it=v[nod].begin();it!=v[nod].end();it++)
if(cap[nod][*it]>flow[nod][*it]&&d[*it]>d[nod]+cost[nod][*it]+dist[nod]-dist[*it])
{
d[*it]=d[nod]+cost[nod][*it]+dist[nod]-dist[*it];
dist1[*it]=dist1[nod]+cost[nod][*it];
tata[*it]=nod;
if(*it!=dest)
h.push({*it,d[*it]});
}
}
for(i=1;i<=n;i++)
dist[i]=dist1[i];
return (d[dest]!=inf);
}
int main()
{
int m,x,y,a,c,sol=0,i;
si>>n>>m>>sursa>>dest;
for(i=1;i<=m;i++)
{
si>>x>>y>>a>>c;
v[x].push_back(y);
v[y].push_back(x);
cap[x][y]=a;
cost[x][y]=c;
cost[y][x]=-c;
}
bf(sursa);
int s;
while(dijkstra())
{
s=inf;
for(i=dest;i!=sursa;i=tata[i])
s=min(s,cap[tata[i]][i]-flow[tata[i]][i]);
for(i=dest;i!=sursa;i=tata[i])
{
flow[tata[i]][i]+=s;
flow[i][tata[i]]-=s;
}
sol+=s*dist[dest];
}
so<<sol;
so.close();
return 0;
}