Cod sursa(job #2025955)

Utilizator tziplea_stefanTiplea Stefan tziplea_stefan Data 23 septembrie 2017 14:41:17
Problema Critice Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.67 kb
#include <fstream>
#include <cstdio>
#include <vector>
#include <queue>
#define VAL 1005
#define MC 10005
#define F first
#define S second
#define INF 1000000000

using namespace std;

int N, M, i, j, C;
int flux[VAL][VAL];
int prec[VAL], X;
bool viz[VAL], ok[2][VAL];
pair <int, int> Mc[MC];
vector <int> G[VAL], SOL;
vector <int> :: iterator it;
queue <int> Q;

bool BFS(int S, int D)
{
    for (j=1; j<=N; j++)
      viz[j]=false;
    viz[S]=true;
    Q.push(S);
    while (Q.empty()==false)
    {
        S=Q.front();
        Q.pop();
        if (S==D)
          continue;
        for (it=G[S].begin(); it!=G[S].end(); it++)
        {
            if (viz[*it]==false && flux[S][*it]>0)
            {
                viz[*it]=true;
                prec[*it]=S;
                Q.push(*it);
            }
        }
    }
    return viz[D]==true;
}

void Ford_Fulkerson()
{
    while (BFS(1, N)==true)
    {
        for (i=0; i<G[N].size(); i++)
        {
            X=INF;
            prec[N]=G[N][i];
            for (j=N; j!=1; j=prec[j])
              X=min(X, flux[prec[j]][j]);
            if (X==0)
              continue;
            for (j=N; j!=1; j=prec[j])
            {
                flux[prec[j]][j]-=X;
                flux[j][prec[j]]+=X;
            }
        }
    }
}

bool Check(pair <int, int> M)
{
    return ((ok[0][M.F]==1 && ok[1][M.S]==1) || (ok[0][M.S]==1 && ok[1][M.F]==1));
}

void BFScrit(int S, int ind)
{
    Q.push(S);
    ok[ind][S]=true;
    while (Q.empty()==false)
    {
        S=Q.front();
        Q.pop();
        for (it=G[S].begin(); it!=G[S].end(); it++)
        {
            if (ind==0)
            {
                if (ok[ind][*it]==false && flux[S][*it]>0)
                {
                    ok[ind][*it]=true;
                    Q.push(*it);
                }
            }
            else
            {
                if (ok[ind][*it]==false && flux[*it][S]>0)
                {
                    ok[ind][*it]=true;
                    Q.push(*it);
                }
            }
        }
    }
}

int main()
{
    freopen("critice.in", "r", stdin);
    freopen("critice.out", "w", stdout);
    scanf("%d %d", &N, &M);
    for (i=1; i<=M; i++)
    {
        scanf("%d %d %d", &Mc[i].F, &Mc[i].S, &C);
        flux[Mc[i].F][Mc[i].S]=C;
        flux[Mc[i].S][Mc[i].F]=C;
        G[Mc[i].F].push_back(Mc[i].S);
        G[Mc[i].S].push_back(Mc[i].F);
    }
    Ford_Fulkerson();
    BFScrit(1, 0);
    BFScrit(N, 1);
    for (i=1; i<=M; i++)
      if ((flux[Mc[i].F][Mc[i].S]==0 || flux[Mc[i].S][Mc[i].F]==0) && Check(Mc[i])==true)
        SOL.push_back(i);
    printf("%d\n", SOL.size());
    for (i=0; i<SOL.size(); i++)
      printf("%d\n", SOL[i]);
    return 0;
}