Cod sursa(job #2822487)

Utilizator OrzataAndreiOrzata Andrei OrzataAndrei Data 24 decembrie 2021 00:06:05
Problema Critice Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.25 kb
#include <bits/stdc++.h>

using namespace std;

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

const int MAXN = 1001;
const int INF  = 10001;

int N,M,x,y,i,j;
int c[MAXN][MAXN],f[MAXN][MAXN],parent[MAXN],visited1[MAXN],visited2[MAXN],q[MAXN];

vector <int>G[MAXN],ANS;
vector <pair<int,int>> edges;

inline int abs(int a)
{
    if(a<0)
        return -a;
    return a;
}

int BFS()
{
    memset(parent,0,sizeof(parent));

    parent[1] = 1;
    q[0] = 1;
    q[1] = 1;

    for (j = 1; j <= q[0]; j++)
    {
        int x = q[j];
        for (int i = 0; i < G[x].size(); i++)
            if (parent[G[x][i]] == 0 && c[x][G[x][i]] > f[x][G[x][i]])
                {
                    parent[G[x][i]] = x;
                    q[++q[0]] = G[x][i];
                }
        if (parent[N])
        return 1;
    }

    return 0;
}

void DFS(int node,int visited[])
{
    visited[node] = 1;
    for(auto x : G[node])
    {
        if(!visited[x] && f[node][x]<c[node][x] && f[x][node]<c[x][node])
        {
            DFS(x,visited);
        }
    }
}

void maxflow()
{
    int fmin = 0;
    while (BFS())
    {
        fmin = INF;
        x = N;
        while (x != 1)
        {
            if(fmin > c[parent[x]][x]-f[parent[x]][x])
                fmin = c[parent[x]][x]-f[parent[x]][x];
            x = parent[x];
        }
        x = N;
        while (x != 1)
        {
            f[parent[x]][x] += fmin;
            f[x][parent[x]] -= fmin;
            x = parent[x];
        }

    }
}

int main()
{

    in>>N>>M;
    for (i = 1; i <= M; i++)
    {
        int x,y,cost;
        in>>x>>y>>cost;
        G[x].push_back(y);
        G[y].push_back(x);

        c[x][y] = cost;
        c[y][x] = cost;
        edges.push_back(make_pair(x,y));
    }

    maxflow();

    DFS(1,visited1);
    DFS(N,visited2);

    for (i = 0; i < M; i++)
        {
            x = edges[i].first;
            y = edges[i].second;
            if ( visited1[x] && visited2[y] || visited1[y] && visited2[x])
                ANS.push_back(i+1);
        }
    out<<ANS.size()<<"\n";
    for (i = 0; i <ANS.size(); i++)
        {
            out<<ANS[i]<<"\n";
        }
    return 0;
}