Cod sursa(job #2693166)

Utilizator AvramDanielAvram Daniel AvramDaniel Data 5 ianuarie 2021 00:23:07
Problema Critice Scor 70
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.97 kb
#include<bits/stdc++.h>
using namespace std;

#define INF 10000
#define N 1001
#define M 10001
#define fi first
#define se second

int path[N];
int adj[N][N];
int n,m;
vector<int> v[N], ans;
pair<int, int> muchii[M];
bool viz[N];
bool bfs(){
    queue<int> q;
    for(int i = 2;i<=n;i++)
        path[i] = 0;

    path[1] = -1;
    q.push(1);
    while(!q.empty() && !path[n]){
        int curr = q.front();
        q.pop();
        for(auto next: v[curr]){
            if(!path[next] && adj[curr][next]){
                path[next] = curr;
                q.push(next);
            }
        }
    }
    return path[n];
}

void maxflow(){
    while(bfs()){
        for(auto pred : v[n]){
            int curr = n;
            path[n] = pred;
            if(path[pred] > 0){
                int flow = INF;
                while(path[curr] != -1){
                    flow = min(flow, adj[path[curr]][curr]);
                    curr = path[curr];
                }
                curr = n;
                while(path[curr] != -1){
                    adj[path[curr]][curr] -= flow;
                    adj[curr][path[curr]] += flow;
                    curr = path[curr];
                }
            }
        }
    }

}

void dfs(int x){
    viz[x] = 1;
    for(auto next: v[x]){
        if(!viz[next] && adj[x][next]){
            dfs(next);
        }
    }
}

int main(){

    ifstream cin("critice.in");
    ofstream cout("critice.out");

    cin>>n>>m;
    for(int i = 1;i<=m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        v[a].push_back(b);
        v[b].push_back(a);
        adj[a][b] = c;
        adj[b][a] = c;
        muchii[i] = {a,b};
    }

    maxflow();

    dfs(1);

    for(int i = 1;i<=m;i++){
        if(viz[muchii[i].fi] != viz[muchii[i].se]){
            ans.push_back(i);
        }
    }

    cout<<ans.size()<<'\n';
    for(auto i: ans){
        cout<<i<<'\n';
    }

    return 0;
}