Cod sursa(job #2975013)

Utilizator PHOSSESSEDProsie Radu-Teodor PHOSSESSED Data 5 februarie 2023 08:17:27
Problema Critice Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.04 kb
#include<fstream>
#include<queue>
#include<vector>
#include<algorithm>
using namespace std;

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

class Graph
{

private:
    struct muchie
    {
        int to,ind,cap,flow,when;
    };
const int INF = 1 << 30;
int n,s,t;
vector<vector<muchie>> vecini;
vector<int> rem,dist;

bool bfs()
{
    fill(dist.begin(),dist.end(),INF);
    dist[s] = 0; queue<int> q; q.push(s);
    while(!q.empty())
        {
            int v = q.front() ; q.pop();
            for(auto it : vecini[v])
                {
                    if(dist[it.to] != (INF)) continue;
                    if(!(it.cap - it.flow)) continue;
                    dist[it.to] = dist[v] + 1;
                    q.push(it.to);
                }
        }

    return dist[t] != (INF);
}

int dfs(int nod,int f)
{
    if(nod == t)
        return f;

    for(int &h = rem[nod] ; h < vecini[nod].size() ; h++)
        {
            muchie &m = vecini[nod][h];
            if(dist[nod] + 1 != dist[m.to]) continue;
            int remcap = m.cap - m.flow;
            if(!remcap) continue;
            int pos = dfs(m.to,min(f,remcap));
            if(pos > 0)
                {
                    m.flow += pos;
                    vecini[m.to][m.ind].flow -= pos;
                    return pos;
                }
        }

    return 0;
}

public:
    Graph(int _n,int _s,int _t)
    {
        n = _n,s = _s,t = _t;
        vecini.resize(n + 1);
        rem.resize(n + 1,0);
        dist.resize(n + 1);
    }

    void add(int a,int b,int c,int timp)
    {
        vecini[a].push_back({b,vecini[b].size(),c,0,timp});
        vecini[b].push_back({a,vecini[a].size() - 1,0,0,timp});
        vecini[b].push_back({a,vecini[a].size(),c,0,timp});
        vecini[a].push_back({b,vecini[b].size() - 1,0,0,timp});
    }

    int maxflow()
    {
        int ans = 0;
        while(bfs())
            {
                fill(rem.begin(),rem.end(),0);
                int delta = dfs(s,INF);
                while(delta != 0)
                    {
                        ans += delta;
                        delta = dfs(s,INF );
                    }
            }

        return ans;
    }

    void getcut()
    {
        bool vfol = bfs();
        vector<int> rez;
        for(int i = 1; i <= n ; i++)
            {
                if(dist[i] == INF) continue;
                for(auto it : vecini[i])
                    {
                        if(dist[it.to] != INF)
                            continue;

                        if(!it.cap) continue;
                        rez.push_back(it.when);
                    }
            }

        cout << rez.size() << '\n';
        for(auto it : rez) cout << it << '\n';
    }

};


int main()
{
    int n,m,a,b,c;
    cin >> n >> m;
    Graph g(n,1,n);
    for(int i = 0 ; i < m ; i++)
        {
            cin >> a >> b >> c;
            g.add(a,b,c,i + 1);
        }

    int ghol = g.maxflow();
    g.getcut();

}