Cod sursa(job #3032762)

Utilizator CristeaCristianCristea Cristian CristeaCristian Data 22 martie 2023 18:08:14
Problema Critice Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.41 kb
#include <fstream>
#include <cstring>
#include <vector>
#include <deque>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <algorithm>
#include <iomanip>
#include <cmath>
#include <cstdlib>
#include <ctime>

using namespace std;
ifstream cin("critice.in");
ofstream cout("critice.out");
using ll = long long;
//#define int ll
using pi = pair<int,int>;
using pll = pair<ll, ll>;

#define pb push_back
#define x first
#define y second

const int MOD = 1e9+7, INF = 1e9, NMAX = 1005;
int cap[NMAX][NMAX], p[NMAX], n, vis[NMAX];
vector<int> v[NMAX];
int bfs(int src, int dest)
{
    int i, nod, u;
    queue<int> q;
    for(i = 1; i <= n; i++)
        p[i] = 0;
    p[src] = src;
    q.push(src);
    while(!q.empty())
    {
        u = q.front();
        q.pop();
        for(i = 0; i < v[u].size(); i++)
        {
            nod = v[u][i];
            if(!p[nod] && cap[u][nod] > 0)
            {
                p[nod] = u;
                q.push(nod);
                if(nod == dest)
                    return 1;
            }
        }
    }
    return 0;
}
void dfs(int u)
{
    int i, nod;
    vis[u] = 1;
    for(i = 0; i < v[u].size(); i++)
    {
        nod = v[u][i];
        if(cap[u][nod] > 0 && !vis[nod])
            dfs(nod);
    }
}
void solve()
{
    int i, a, b, c, src, dest, new_flow, u, m, flow = 0;
    vector<pi> edges;
    cin >> n >> m;
    for(i = 1; i <= m; i++)
    {
        cin >> a >> b >> c;
        cap[a][b] = c;
        cap[b][a] = c;
        v[a].pb(b);
        v[b].pb(a);
        edges.pb({a, b});
    }
    src = 1;
    dest = n;
    while(bfs(src, dest))
    {
        new_flow = INF;
        u = dest;
        while(u != src)
        {
            new_flow = min(new_flow, cap[p[u]][u]);
            u = p[u];
        }
        u = dest;
        while(u != src)
        {
            cap[p[u]][u] -= new_flow;
            cap[u][p[u]] += new_flow;
            u = p[u];
        }
        flow += new_flow;
    }
    dfs(src);
    vector<int> ans;
    for(i = 0; i < m; i++)
        if(vis[edges[i].x] + vis[edges[i].y] == 1)
            ans.pb(i + 1);
    cout << ans.size() << '\n';
    for(auto it: ans)
        cout << it << '\n';
}
int main()
{
    cin.tie(0)->sync_with_stdio(0);
    cin.exceptions(cin.failbit);
    int T;
    T = 1;
    while(T--)
        solve();
    return 0;
}