Pagini recente » Cod sursa (job #3131899) | Cod sursa (job #3195477) | Cod sursa (job #1928988) | Cod sursa (job #1212647) | Cod sursa (job #2288308)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
using namespace std;
const int maxn = 1005, maxm = 5005, inf = 0x3f3f3f3f;
ifstream f("critice.in");
ofstream g("critice.out");
int n, m, i, ans;
vector <int> gr[maxn];
int c[maxn][maxn], us[maxn][maxn]; // cost, used
int p[maxn]; bool is[maxn];
queue <int> q;
int rez[maxm], rezk;
bool bfs()
{
memset(is, false, sizeof(is));
while(!q.empty()) {
q.pop();
}
q.push(1);
is[1] = true;
while(!q.empty())
{
int x = q.front();
q.pop();
if(x == n) { continue; }
for(auto u : gr[x])
{
if(!(is[u] == true || us[x][u] == c[x][u])) {
p[u] = x;
q.push(u);
is[u] = true;
}
}
}
return is[n];
}
int maxflow()
{
ans = 0;
for(int i = 1; i <= n; i ++) {
for(int j = 1; j <= n; j ++) {
us[i][j] = 0;
}
}
while(bfs())
{
for(auto u : gr[n])
{
if(us[u][n] == c[u][n] || is[u] == false) {
continue;
}
p[n] = u;
int minim = inf;
for(int nod = n; nod != 1; nod = p[nod]) {
minim = min(minim, c[p[nod]][nod] - us[p[nod]][nod]);
}
for(int nod = n; nod != 1; nod = p[nod]) {
us[p[nod]][nod] += minim;
us[nod][p[nod]] -= minim;
}
ans += minim;
}
}
return ans;
}
struct xy {
int x, y;
};
xy v[maxm];
int main()
{
int x, y, z;
f >> n >> m;
for(i = 1; i <= m ; i++)
{
f >> x >> y >> z;
v[i] = {x, y};
gr[x].push_back(y);
gr[y].push_back(x);
c[x][y] = z;
c[y][x] = z;
}
int initial = maxflow();
for(i = 1; i <= m; i ++)
{
int aux = c[v[i].x][v[i].y];
c[v[i].x][v[i].y] = inf;
c[v[i].y][v[i].x] = inf;
int now = maxflow();
if(now > initial) {
rez[++rezk] = i;
}
c[v[i].x][v[i].y] = aux;
c[v[i].y][v[i].x] = aux;
}
g << rezk << '\n';
for(i = 1; i <= rezk; i ++) {
g << rez[i] << '\n';
}
f.close();
g.close();
return 0;
}