Pagini recente » Cod sursa (job #914020) | Cod sursa (job #982584) | Cod sursa (job #337128) | Cod sursa (job #1137725) | Cod sursa (job #712223)
Cod sursa(job #712223)
#include <fstream>
#include <vector>
#include <cstring>
#include <bitset>
#include <queue>
#define st first
#define nd second
#define PSS pair<short,short>
using namespace std;
ifstream f("fmcm.in");
ofstream g("fmcm.out");
const short MAXN = 355 , INF = 32000;
vector< PSS > G[MAXN];
short S , D , N , M , drum;
short Dist[MAXN] , C[MAXN][MAXN] , F[MAXN][MAXN] , TT[MAXN];
bitset<MAXN> u;
struct cmp { bool operator() (const short &x,const short &y)
const { return &Dist[x] > &Dist[y]; }
};
priority_queue<short,vector<short>,cmp > Q;
void read_data()
{
int x , y , cap ,z;
for(f>>N>>M>>S>>D;M;M--)
{
f>>x>>y>>cap>>z;
G[x].push_back(make_pair(y,z));
G[y].push_back(make_pair(x,-z));
C[x][y] = cap;
}
}
int bellmanford()
{
short v , fmin = INF;
fill(Dist + 1,Dist + N + 1,INF);
u.reset();
Dist[S] = 0; u[S] = 1;
Q.push(S);
while(!Q.empty())
{
v = Q.top() , Q.pop();
for(vector< PSS >::const_iterator w=G[v].begin();w!=G[v].end();++w)
if(C[v][w->st] > F[v][w->st] && w->nd + Dist[v]<Dist[w->st])
{
Dist[w->st] = w->nd + Dist[v];
TT[w->st] = v;
if(u[w->st] == 0)
u[w->st] = 1 , Q.push(w->st);
}
u[v] = 0;
}
if(Dist[D]!=INF)
{
drum = 1;
for(int i=D;i!=S;i=TT[i])
if(C[TT[i]][i] - F[TT[i]][i] < fmin) fmin = C[TT[i]][i] - F[TT[i]][i];
for(int i=D;i!=S;i=TT[i])
F[TT[i]][i]+=fmin , F[i][TT[i]]-=fmin;
return Dist[D]*fmin;
}
return 0;
}
int solve()
{
int ans = 0;
do
{ drum = 0;
ans+=bellmanford();
} while(drum);
return ans;
}
int main()
{
read_data();
g<<solve()<<'\n';
return 0;
}