Pagini recente » Cod sursa (job #584511) | Cod sursa (job #2901896) | Cod sursa (job #827790) | Cod sursa (job #1437618) | Cod sursa (job #1159973)
#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;
}