Cod sursa(job #2725148)

Utilizator retrogradLucian Bicsi retrograd Data 18 martie 2021 14:55:41
Problema Cuplaj maxim de cost minim Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.1 kb
#include <bits/stdc++.h>
 
using namespace std;
using ll = int;
const ll INF = 1e9;
 
struct ZKW {
  struct Edge { int a, b; ll f, c, k; };
  int n;
  vector<Edge> E;
  vector<vector<int>> graph;
  vector<ll> pi; vector<int> at, vis;
 
  ZKW(int n) : n(n), graph(n), at(n, 0), pi(n, 0) {}
 
  void AddEdge(int a, int b, ll c, ll k) {
    graph[a].push_back(E.size()); 
    E.push_back({a, b, 0, c, k}); 
    graph[b].push_back(E.size()); 
    E.push_back({b, a, 0, 0, -k}); 
  }
  bool relabel() {
    ll upd = INF;
    for (auto& e : E) if (vis[e.a] && !vis[e.b] && e.f < e.c)
      upd = min(upd, pi[e.a] + e.k - pi[e.b]);
    assert(upd >= 0);
    for (int i = 0; i < n; ++i) if (!vis[i]) pi[i] += upd;
    at.assign(n, 0);
    return upd != INF;
  }
  ll dfs(int v, int t, ll flow) {
    vis[v] = true;
    if (v == t) return flow;
    for (int& i = at[v]; i < (int)graph[v].size(); ++i) {
      int ei = graph[v][i]; auto& e = E[ei]; ll ret;
      if (pi[e.b] == pi[e.a] + e.k && e.f < e.c && !vis[e.b] &&
          (ret = dfs(e.b, t, min(flow, e.c - e.f)))) 
        return E[ei].f += ret, E[ei^1].f -= ret, ret; 
    }
    return 0;
  }
  pair<ll, ll> Compute(int s, int t) {
    ll flow = 0, cost = 0, f;
    while (true) {
      vis.assign(n, 0);
      if (!(f = dfs(s, t, INF)) && !relabel()) break;
      flow += f; cost += f * pi[t];
    }
    return {flow, cost};
  }
  void SetPi(int s) {
    for (int ch = 1; ch--; ) 
      for (auto& e : E) 
        if (e.f < e.c && pi[e.b] > pi[e.a] + e.k)
          pi[e.b] = pi[e.a] + e.k, ch = 1;
  }
};
 
int main() {
  ifstream cin("cmcm.in");
  ofstream cout("cmcm.out");
  int n, m, e; cin >> n >> m >> e;
  ZKW F(n + m + 2);
  for (int i = 0; i < e; ++i) {
    int a, b, c; cin >> a >> b >> c;
    F.AddEdge(a - 1, b + n - 1, 1, c);
  }
  for (int i = 0; i < n; ++i)
    F.AddEdge(n + m, i, 1, 0);
  for (int i = 0; i < m; ++i)
    F.AddEdge(i + n, n + m + 1, 1, 0);
  F.SetPi(n + m);
  auto p = F.Compute(n + m, n + m + 1);
  cout << p.first << " " << p.second << endl;
  for (int i = 0; i < e; ++i) {
    if (F.E[2 * i].f > 0)
      cout << i + 1 << " ";
  }
  cout << endl;
  return 0;
}