Cod sursa(job #1159460)

Utilizator manutrutaEmanuel Truta manutruta Data 29 martie 2014 17:01:51
Problema Critice Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 3.57 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];
pair<int, int> muchie[MAXM];
vector< pair<int, int> > G[MAXN];
bitset<MAXN> viz;
bitset<MAXM> vizmuchie, vizmuchie2;
queue<int> q;

bool bfs()
{
    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)
{
    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)
{
    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);
    }
}

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;
        }
    }

    dfs1(1);
    dfs2(n);

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

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

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