Cod sursa(job #1159973)

Utilizator manutrutaEmanuel Truta manutruta Data 29 martie 2014 23:57:34
Problema Critice Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.95 kb
#include <iostream>
#include <fstream>
#include <bitset>
#include <vector>
#include <queue>

using namespace std;
typedef vector< pair<int, int> >::iterator iter;

#define INF 0x3f3f3f3f
#define MAXN 1005
#define MAXM 10005

// <parsing>
FILE* f = fopen("critice.in", "r");
unsigned const maxb = 8192;
char buf[maxb];
int ptr = maxb;

inline int getInt()
{
    while (buf[ptr] < '0' || '9' < buf[ptr]) {
        if (++ptr >= maxb) {
            fread(buf, maxb, 1, f);
            ptr = 0;
        }
    }
    int nr = 0;
    while ('0' <= buf[ptr] && buf[ptr] <= '9') {
        nr = nr * 10 + buf[ptr] - '0';
        if (++ptr >= maxb) {
            fread(buf, maxb, 1, f);
            ptr = 0;
        }
    }
    return nr;
}
// </parsing>
ofstream g("critice.out");

int n;
int cap[MAXN][MAXN], fl[MAXN][MAXN], t[MAXN], bfl[MAXN][MAXN];
pair<int, int> muchie[MAXM];
vector< pair<int, int> > G[MAXN];
bitset<MAXN> viz;
int viznod[MAXN];
bitset<MAXM> vizmuchie, vizmuchie2;
queue<int> q;

bool bfs()
{
    for (int i = 1; i <= n; i++) {
        viznod[i] = 0;
    }
    while (!q.empty()) q.pop();
    viz.reset();
    viz[0] = true;

    q.push(1);
    while (!q.empty()) {
        int nd = q.front();
        q.pop();

        for (iter it = G[nd].begin(); it != G[nd].end(); it++) {
            if (viz[it->first] == true || cap[nd][it->first] == fl[nd][it->first]) {
                continue;
            }

            viz[it->first] = true;
            t[it->first] = nd;
            q.push(it->first);
        }
    }

    return viz[n];
}

void dfs1(int nd)
{
    if (viz[nd] == true) {
        return ;
    }
    viz[nd] = true;

    for (iter it = G[nd].begin(); it != G[nd].end(); it++) {
        if (vizmuchie[it->second] == true) {
            continue;
        }
        vizmuchie[it->second] = true;

        if (cap[nd][it->first] == fl[nd][it->first]) {
            continue;
        }
        dfs1(it->first);
    }
}

void dfs2(int nd)
{
    if (viz[nd] == true) {
        return ;
    }
    viz[nd] = true;

    for (iter it = G[nd].begin(); it != G[nd].end(); it++) {
        if (vizmuchie2[it->second] == true) {
            continue;
        }
        vizmuchie2[it->second] = true;

        if (cap[it->first][nd] == fl[it->first][nd]) {
            continue;
        }
        dfs2(it->first);
    }
}

bool saturata(int x, int y)
{
    if (cap[x][y] == max(fl[x][y], fl[y][x])) {
        return true;
    }
    return false;
}

int main()
{
    int m;
    n = getInt(); m = getInt();
    for (int i = 1; i <= m; i++) {
        int x, y, c;
        x = getInt(); y = getInt(); c = getInt();
        cap[x][y] = c;
        cap[y][x] = c;

        G[x].push_back(make_pair(y, i));
        G[y].push_back(make_pair(x, i));
        muchie[i].first = x;
        muchie[i].second = y;
    }

    int flow = 0;
    while (bfs()) {
        for (iter it = G[n].begin(); it != G[n].end(); it++) {
            if (viz[it->first] == false) {
                continue;
            }
            t[n] = it->first;

            int fmin = INF;
            for (int i = n; i != 1; i = t[i]) {
                fmin = min(fmin, cap[t[i]][i] - fl[t[i]][i]);
            }
            for (int i = n; i != 1; i = t[i]) {
                fl[t[i]][i] += fmin;
                fl[i][t[i]] -= fmin;
            }
            flow += fmin;
        }
    }

    viz.reset();
    dfs1(1);
    viz.reset();
    dfs2(n);

    int sol = 0;
    for (int i = 1; i <= m; i++) {
        if (vizmuchie[i] == 1 && vizmuchie2[i] == 1 && saturata(muchie[i].first, muchie[i].second)) {
            sol++;
        }
    }

    g << sol << '\n';
    for (int i = 1; i <= m; i++) {
        if (vizmuchie[i] == 1 && vizmuchie2[i] == 1 && saturata(muchie[i].first, muchie[i].second)) {
            g << i << '\n';
        }
    }

    fclose(f);
    g.close();
    return 0;
}