Cod sursa(job #1109578)

Utilizator stefanzzzStefan Popa stefanzzz Data 17 februarie 2014 13:06:37
Problema Critice Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <fstream>
#include <vector>
#include <algorithm>
#define MAXN 1005
#define INF 2000000000
using namespace std;
ifstream f("critice.in");
ofstream g("critice.out");

int n,m,a,b,x,mn;
int c[MAXN][MAXN],nrm[MAXN][MAXN],flux[MAXN][MAXN],tata[MAXN];
vector<int> G[MAXN],sol;
bool uz[MAXN],uzn[MAXN];

bool DFS(int p);

int main()
{
    int i,j,k;
    f>>n>>m;
    for(i=1;i<=m;i++){
        f>>a>>b;
        G[a].push_back(b);
        G[b].push_back(a);
        f>>c[a][b];
        c[b][a]=c[a][b];
        nrm[a][b]=nrm[b][a]=i;}
    DFS(1);
    while(uz[n]){
        x=n;
        mn=INF;
        while(tata[x]){
            if(c[tata[x]][x]-flux[tata[x]][x]<mn)
                mn=c[tata[x]][x]-flux[tata[x]][x];
            x=tata[x];}
        x=n;
        while(tata[x]){
            flux[tata[x]][x]+=mn;
            flux[x][tata[x]]-=mn;
            x=tata[x];}
        for(i=1;i<=n;i++)
            uz[i]=0;
        DFS(1);}
    for(i=1;i<=n;i++)
        uz[i]=0;
    DFS(n);
    for(i=1;i<=n;i++){
        uzn[i]=uz[i];
        uz[i]=0;}
    DFS(1);
    for(i=1;i<=n;i++)
        for(j=1;j<=n;j++)
            if(c[i][j]&&flux[i][j]==c[i][j]&&uz[i]&&uzn[j])
                sol.push_back(nrm[i][j]);
    sort(sol.begin(),sol.end());
    g<<sol.size()<<'\n';
    for(i=0;i<sol.size();i++)
        g<<sol[i]<<'\n';
    f.close();
    g.close();
    return 0;
}

bool DFS(int p){
    int i;
    uz[p]=1;
    for(i=0;i<G[p].size();i++){
        x=G[p][i];
        if(!uz[x]&&flux[p][x]<c[p][x]){
            tata[x]=p;
            DFS(x);}}}