Cod sursa(job #1051203)

Utilizator narcis_vsGemene Narcis - Gabriel narcis_vs Data 9 decembrie 2013 20:12:01
Problema Critice Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.46 kb
#include <fstream>
#include <queue>
#include <bitset>

using namespace std;
struct edge
{
    int x, y, c;
    edge()
    {
        x = y = c = 0;
    }
    inline edge(const int _x,const int _y,const int _c)
    {
        x = _x,    y = _y,     c = _c;
    }
};
const int Nmax = 1004;
const int Mmax =10004;
int C[Nmax][Nmax];
int F[Nmax][Nmax];
int R[Nmax][Nmax];
int n, m, Father[Nmax];
edge e[Mmax];
bitset < Nmax > viz,a,b;
vector < int > sol;
inline void Read()
{
    ifstream f("critice.in");
    f >> n >> m;
    int x, y, c;
    for(int i = 1;i <= m; ++i)
    {
        f >> x >> y >> c;
        C[x][y] = C[y][x] = c;
        e[i] = edge(x,y,c);
    }
}

inline bool BFS(int node)
{
    queue < int > Q;
    viz = 0;
    viz[node] = 1;
    register int i;
    for(Q.push(node); !Q.empty(); Q.pop())
    {
        node = Q.front();
        if(node == n)
            continue ;
        for(i = 1;i <= n; ++i)
            if(!viz[i] && F[node][i] < C[node][i])
            {
                Q.push(i);
                viz[i] = 1;
                Father[i] = node;
            }
    }
    return viz[n];
}

inline void MaxFlow()
{
    int i,j,minn ;
    while(BFS(1))
    {
        for(j = 1; j < n; ++j)
        {
            if(!viz[j] || !C[j][n] || C[j][n] == F[j][n])
                continue ;
            minn = 70004;
            Father[n] = j;
            for(i = n;i != 1;i = Father[i])
                minn = min(minn,C[Father[i]][i]-F[Father[i]][i]);
            for(i = n;i != 1;i = Father[i])
                F[Father[i]][i]+= minn,
                F[i][Father[i]] -= minn;
        }
    }
}
inline void DFS1(const int node)
{
    a[node] = 1;
    for(int i = 1;i <= n; ++i)
        if(F[node][i] < C[node][i] && !a[i])
            DFS1(i);
}
inline void DFS2(const int node)
{
    b[node] = 1;
    for(int i = 1;i <= n; ++i)
        if(F[i][node] < C[i][node] && !b[i])
            DFS2(i);
}
inline void Solve()
{
    int x,y,c,i;
    DFS1(1);
    DFS2(n);
    for(i = 1;i <= m; ++i)
    {
        x = e[i].x,  y = e[i].y,  c = e[i].c;
        if(F[x][y]==c && a[x] && b[y])
            sol.push_back(i);
        else
        {
            swap(x,y);
            if(F[x][y]==c && a[x] && b[y])
                sol.push_back(i);
        }
    }
}

inline void Write()
{
    int i,n;
    ofstream g("critice.out");
    g<<( n = sol.size() )<<"\n";
    for(i = 0;i < n; ++i)
        g<<sol[i]<<"\n";
    g.close();
}


int main()
{
    Read();
    MaxFlow();
    Solve();
    Write();
    return 0;
}