Cod sursa(job #2663970)

Utilizator divianegoescuDivia Negoescu divianegoescu Data 27 octombrie 2020 18:19:00
Problema Critice Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.18 kb
#include <fstream>
#include <vector>
#include <bitset>
#define K 1005
using namespace std;
ifstream fin("critice.in");
ofstream fout("critice.out");
int n,m,i,j,x,y,z,c[K][K],f[K][K],t[K],q[K],p,u,Left[K],Right[K];
bitset <K> viz;
vector <int> v[K],sol;
pair <int,int> muchii[K*10];
int bfs(){
    viz.reset();
    q[1]=viz[1]=1;
    for(p=u=1;p<=u;p++){
        int nod=q[p];
        for(auto vecin: v[nod])
            if(!viz[vecin] && c[nod][vecin]>f[nod][vecin]){
                q[++u]=vecin;
                t[vecin]=nod;
                viz[vecin]=1;
                if(vecin==n)
                    return 1;
            }
    }
    return 0;
}
int mod(int x){
    if(x<0)return -x;
    return x;
}
void verif(int sursa,int ok[]){
    viz.reset();
    viz[sursa]=ok[sursa]=1;
    q[1]=sursa;
    for(p=u=1;p<=u;p++){
        int nod=q[p];
        for(auto vecin: v[nod]){
            if(!viz[vecin] && c[nod][vecin]>mod(f[nod][vecin])){
                viz[vecin]=ok[vecin]=1;
                q[++u]=vecin;
            }
        }
    }
}
int main(){
    fin>>n>>m;
    for(i=1;i<=m;i++){
        fin>>x>>y>>z;
        v[x].push_back(y);
        v[y].push_back(x);
        c[x][y]=c[y][x]=z;
        muchii[i]={x,y};
    }
    while(bfs())
        for(auto vecin: v[n]){
            int scad=c[vecin][n]-f[vecin][n];
            if(viz[vecin] && scad>0){
                x=vecin;
                while(t[x]){
                    scad=min(scad,c[t[x]][x]-f[t[x]][x]);
                    x=t[x];
                }
                f[vecin][n]+=scad;
                f[n][vecin]-=scad;
                x=vecin;
                while(t[x]){
                    f[t[x]][x]+=scad;
                    f[x][t[x]]-=scad;
                    x=t[x];
                }
            }
        }
    verif(1,Left); //L[i]=1 daca e posibil 1->x pe much nesaturate
    verif(n,Right); // n->x
    for(i=1;i<=m;i++){
        x=muchii[i].first;
        y=muchii[i].second;
        if((Left[x]&&Right[y]) || (Left[y]&&Right[x]))
            sol.push_back(i);
    }
    fout<<sol.size()<<"\n";
    for(auto i:sol)
        fout<<i<<"\n";
    return 0;
}