Cod sursa(job #2085595)

Utilizator MiricaMateiMirica Matei MiricaMatei Data 10 decembrie 2017 14:24:07
Problema Cuplaj maxim de cost minim Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.33 kb
#include <bits/stdc++.h>

using namespace std;

const int INF = (1LL << 31) - 1;
const int MAX_N = 602;

vector<int>G[MAX_N + 5];
int nrv[MAX_N + 5];
bool inc[MAX_N + 5];
int dist[MAX_N + 5];
int from[MAX_N + 5];
int cr[MAX_N + 5][MAX_N + 5];
int cost[MAX_N + 5][MAX_N + 5];
int muchie[MAX_N + 5][MAX_N + 5];

int bellman(int s, int d, int n) {
  for (int i = 1; i <= n; ++i) {
    dist[i] = INF;
    nrv[i] = 0;
    inc[i] = 0;
  }
  inc[s] = 1;
  from[s] = s;
  nrv[s] = 1;
  dist[s] = 0;
  queue<int>q;
  q.push(s);
  while (!q.empty() && nrv[q.front()] < n) {
    int node = q.front();
    q.pop();
    inc[node] = 0;
    for (auto it:G[node]) {
      if (cr[node][it] > 0 && dist[node] + cost[node][it] < dist[it]) {
        from[it] = node;
        dist[it] = dist[node] + cost[node][it];
        if (inc[it] == 0) {
          inc[it] = 1;
          q.push(it);
          nrv[it]++;
        }
      }
    }
  }
  return dist[d];
}

void cmcm(int s, int d, int n) {
  int ans1 = 0, ans2 = 0, dm;
  dm = bellman(s, d, n);
  while (dm != INF) {
    int aux = INF;
    int node = d;
    while (node != s) {
      aux = min(aux, cr[from[node]][node]);
      node = from[node];
    }
    node = d;
    while (node != s) {
      cr[from[node]][node] -= aux;
      cr[node][from[node]] += aux;
      node = from[node];
    }
    ans1 += aux;
    ans2 += dm * aux;
    dm = bellman(s, d, n);
  }

  printf("%d %d\n", ans1, ans2);
}

int main() {
  freopen("cmcm.in", "r", stdin);
  freopen("cmcm.out", "w", stdout);

  int l, r, m;
  scanf("%d%d%d", &l, &r, &m);
  for (int i = 1; i <= m; ++i) {
    int u, v, cst;
    scanf("%d%d%d", &u, &v, &cst);
    u++;
    v = v + l + 1;
    G[u].push_back(v);
    G[v].push_back(u);
    cr[u][v] = 1;
    muchie[u][v] = i;
    cost[u][v] = cst;
    cost[v][u] = -cst;
  }
  for (int i = 2; i <= l + 1; ++i) {
    cr[1][i] = 1;
    G[1].push_back(i);
    G[i].push_back(1);
  }
  for (int i = l + 2; i <= l + r + 1; ++i) {
    cr[i][l + r + 2] = 1;
    G[i].push_back(l + r + 2);
    G[l + r + 2].push_back(i);
  }

  cmcm(1, l + r + 2, l + r + 2);
  for (int i = 2; i <= l + 1; ++i)
    for (int j = l + 2; j <= l + r + 1; ++j)
      if (muchie[i][j] && cr[i][j] == 0)
        printf("%d ", muchie[i][j]);
  printf("\n");

  return 0;
}