Cod sursa(job #2697207)

Utilizator jucatorulGrigore George Alexandru jucatorul Data 17 ianuarie 2021 19:49:05
Problema Critice Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.25 kb
#include <bits/stdc++.h>

using namespace std;

vector<int> a[10001];
int c[10001][10001];
int f[10001][10001];
int qu[10001];
int is[2][10001];
int sol[100001];
int viz[10001];
int st[100001];
int dr[100001];
int t[10001];
int N,M;

int bfs()
{
    int i,x,y,inc,sf;

    memset(t, 0, sizeof(t));
    inc = sf = 0;
    qu[0] = 1;
    while(inc <= sf)
    {
        x = qu[inc++];
        for(i = 0; i < a[x].size(); ++i)
        {
            y = a[x][i];
            if(!t[y] && f[x][y] < c[x][y])
            {
                qu[++sf] = y;
                t[y] = x;
            }
        }
    }

    return t[N];
}

void cale(int S, int l)
{
    int i,x,y,inc,sf,sgn=1;

    memset(viz, 0, sizeof(viz));
    if(l) sgn = -1;
    is[l][S] = 1;
    inc = sf = 0;
    qu[0] = S;
    while(inc <= sf)
    {
        x = qu[inc++];
        for(i = 0; i < a[x].size(); ++i)
        {
            y = a[x][i];
            if(!viz[y] && sgn * f[x][y] >= 0 && abs(f[x][y]) < c[x][y])
            {
                is[l][y] = viz[y] = 1;
                qu[++sf] = y;
            }
        }
    }
}

int main(){

 ifstream f("critice.in");
 ofstream g("critice.out");
    int i,j,x,y,z,r,cnt=0;

   f>>N>>M;
    for(i = 1; i <= M; ++i)
    {
      f>>x>>y>>z;
        st[i] = x; dr[i] = y;
        c[x][y] = c[y][x] = z;
        a[x].push_back(y);
        a[y].push_back(x);
    }

    while(bfs())
        for(i = 0; i < a[N].size(); ++i)
        {
            x = a[N][i];
            if(!t[x] || c[x][N] == f[x][N]) continue;

            r = c[x][N] - f[x][N];
            for(j = x; j > 1; j = t[j]) r = min(r, c[ t[j] ][j] - f[ t[j] ][j]);
            if(!r) continue;

            f[x][N] += r;
            f[N][x] -= r;
            for(j = x; j > 1; j = t[j])
            {
                f[ t[j] ][j] += r;
                f[j][ t[j] ] -= r;
            }
        }

    cale(1,0);
    cale(N,1);
    for(i = 1; i <= M; ++i)
    {
        x = st[i];
        y = dr[i];
        if(f[x][y] < 0) swap(x,y);
        if(f[x][y] == c[x][y])
            if(is[0][x] && is[1][y]) ++cnt, sol[i] = 1;
    }
g<<cnt<<"\n";

    for(i = 1; i <= M; ++i)
        if(sol[i])g<<i<<"\n";


    return 0;
}