Cod sursa(job #2386005)

Utilizator MateiTrandafirMatei Trandafir MateiTrandafir Data 22 martie 2019 00:51:09
Problema Critice Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.32 kb
#include <fstream>
#include <cstring>
#include <algorithm>

constexpr int MAX_N = 1001, MAX_M = 20001, INF = 0x7fffffff;

int n, m;
int first[MAX_N], start[MAX_N], id[MAX_M], next[MAX_M], at[MAX_M], count;
int c[MAX_N][MAX_N], f[MAX_N][MAX_N];

inline void add(int from, int to, int index) {
    ++count;
    at[count] = to;
    id[count] = index;
    next[count] = first[from];
    first[from] = count;
}

int q[MAX_N], level[MAX_N], st, dr;

inline bool bfs() {
    std::memset(level, 0xff, sizeof(level));
    st = dr = 0;
    q[0] = 1;
    level[1] = 0;
    int p, x, y;
    while (st <= dr) {
        x = q[st++];
        for (p = first[x]; p != 0; p = next[p]) {
            y = at[p];
            if (level[y] == -1 && f[x][y] < c[x][y]) {
                level[y] = level[x] + 1;
                q[++dr] = y;
            }
        }
    }
    return level[n] != -1;
}

inline int flux(int x, int fMin) {
    if (x == n) return fMin;
    int y;
    for (; start[x] != 0; start[x] = next[start[x]]) {
        y = at[start[x]];
        if (level[y] == level[x] + 1 && f[x][y] < c[x][y]) {
            fMin = std::min(fMin, c[x][y] - f[x][y]);
            fMin = flux(y, fMin);
            if (fMin > 0) {
                f[x][y] += fMin;
                f[y][x] -= fMin;
                return fMin;
            }
        }
    }
    return 0;
}

int r[MAX_M], rCount;

void rBfs() {
    std::memset(level, 0xff, sizeof(level));
    st = dr = 0;
    q[0] = 1;
    level[1] = 0;
    int p, x, y;
    while (st <= dr) {
        x = q[st++];
        for (p = first[x]; p != 0; p = next[p]) {
            y = at[p];
            if (f[x][y] == c[x][y] || f[y][x] == c[y][x]) r[rCount++] = id[p];
            else if (level[y] == -1) {
                level[y] = level[x] + 1;
                q[++dr] = y;
            }
        }
    }
}

int main() {
    std::ifstream in("critice.in");
    std::ofstream out("critice.out");
    int i, x, y, z;
    in >> n >> m;
    for (i = 1; i <= m; ++i) {
        in >> x >> y >> z;
        add(x, y, i);
        add(y, x, i);
        c[x][y] = c[y][x] = z;
    }
    int fc;
    while (bfs()) {
        for (i = 1; i <= n; ++i) start[i] = first[i];
        do fc = flux(1, INF);
        while (fc > 0);
    }
    rBfs();
    std::sort(r, r + rCount);
    out << rCount;
    for (i = 0; i < rCount; ++i) out << '\n' << r[i];
    return 0;
}