Pagini recente » Cod sursa (job #2359468) | Cod sursa (job #954157) | Cod sursa (job #864917) | Cod sursa (job #1927035) | Cod sursa (job #1447794)
#include <fstream>
#include <vector>
#include <queue>
using namespace std;
ifstream fin("cmcm.in");
ofstream fout("cmcm.out");
const int NSIZE = 2 * 300 + 10;
const int inf = 1 << 30;
int N , M , E , S , D , i , j , x , y , z , fmin , flow , hookUp;
int c[NSIZE][NSIZE] , f[NSIZE][NSIZE] , label[NSIZE][NSIZE] , p[NSIZE][NSIZE];
int nxt[NSIZE] , d[NSIZE] , flag[NSIZE];
vector < int > g[NSIZE];
queue < int > inQ;
void add(int x,int y,int z)
{
g[x].push_back(y);
g[y].push_back(x);
c[x][y] = 1;
p[x][y] = z;
p[y][x] = -z;
}
bool bellman()
{
for (int i=S;i<=D;++i)
d[i] = inf;
d[S] = 0;
inQ.push(S);
flag[S] = 1;
while (!inQ.empty())
{
int crt = inQ.front();
inQ.pop() , flag[crt] = 0;
for (int i = 0;i < g[crt].size();++i)
{
int to = g[crt][i];
if (f[crt][to] < c[crt][to] && d[to] > d[crt] + p[crt][to])
{
d[to] = d[crt] + p[crt][to];
nxt[to] = crt;
if (flag[to] == 0)
{
inQ.push(to);
flag[to] = 1;
}
}
}
}
if (d[D] == inf) return false;
return true;
}
int main()
{
fin >> N >> M >> E;
S = 0 , D = N + M + 1;
for (i=1;i<=E;++i)
{
fin >> x >> y >> z;
label[x][y] = i;
add(x , y + N , z);
}
for (i=1;i<=N;++i)
add(S , i , 0);
for (i=1;i<=M;++i)
add(i + N , D , 0);
while (bellman())
{
fmin = inf;
for (i = D ; i != S ; i = nxt[i])
{
int to = nxt[i];
fmin = min(fmin , c[to][i] - f[to][i]);
}
flow += fmin * d[D];
hookUp += fmin;
for (i = D ; i != S ; i = nxt[i])
{
int to = nxt[i];
f[to][i] += fmin;
f[i][to] -= fmin;
}
}
fout << hookUp << " " << flow << '\n';
for (i=1;i<=N;++i)
for (j=1;j<=M;++j)
if (f[i][j + N])
fout << label[i][j] << " ";
fout << '\n';
return 0;
}