Cod sursa(job #3336933)

Utilizator uncrownedHojda Adelin uncrowned Data 26 ianuarie 2026 18:58:34
Problema Critice Scor 30
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.3 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("critice.in");
ofstream fout("critice.out");

#define cin fin 
#define cout fout

vector<int> g[1001];
int c[1001][1001];
int flow[1001][1001];
int parent[1001];
int n, m;
bool v[1001];
vector<pair<int,int>> edge;
bool bfs() {
    queue<int> q;
    q.push(1);
    memset(v,0,sizeof(v));
    v[1]=true;
    while(!q.empty()) {
        int node = q.front();
        q.pop();
        if (node==n) continue;
        for(auto i : g[node]) {
            if (flow[node][i]==c[node][i] || v[i]) continue;
            v[i]=true;
            q.push(i);
            parent[i]=node;
        }
    }
    return v[n];
}

int main()
{
    cin >> n >> m;
    int x, y, cap;
    for(int i = 1; i <= m; i++) {
        cin >> x >> y >> cap;
        c[x][y]+=cap;
        g[x].push_back(y);
        g[y].push_back(x);
        edge.push_back({x, y});
    }
    int ans=0;
    int av[1001];
    int maxflow=0;
    for(; bfs(); ) {
        for(auto node : g[n]) {
            if (flow[node][n]==c[node][n] || !v[node]) continue;
            parent[n]=node;
            int flowmin=INT_MAX;
            for(node=n; node != 1; node=parent[node]) {
                flowmin = min(flowmin, c[parent[node]][node]-flow[parent[node]][node]);
            }
            if (flowmin==0) continue;
            for(node=n; node != 1; node=parent[node]) {
                flow[parent[node]][node] += flowmin;
                flow[node][parent[node]] -= flowmin;
            }
            maxflow += flowmin;
        }
    }
    memset(v,0,sizeof(v));
    queue<int> q;
    q.push(1);
    v[1]=true;
    while(!q.empty()) {
        int node = q.front();
        q.pop();
        for(auto i : g[node]) {
            if (v[i]||flow[node][i]==c[node][i]) continue;
            q.push(i);
            v[i]=true;
        }
    }
    for(int i = 0; i < edge.size(); i++) {
        int x = edge[i].first, y = edge[i].second;
        if (v[x]&&!v[y]&&c[x][y]==flow[x][y]) {
            av[++ans]=i+1;
        }
        if (v[y]&&!v[x]&&c[y][x]==flow[y][x]) {
            av[++ans]=i+1;
        }
    }
    cout << ans << "\n";
    sort(av+1, av+1+ans);
    for(int i = 1; i <= ans; i++) {
        cout << av[i] << "\n";
    }
    
    return 0;
}