Cod sursa(job #1934830)

Utilizator geo_furduifurdui geo geo_furdui Data 21 martie 2017 20:58:57
Problema Critice Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.42 kb
#include <fstream>
#include<vector>
#include<queue>
#include<climits>
using namespace std;
ifstream cin ("critice.in");
ofstream cout ("critice.out");
vector <int> t[1010];
queue <int> coada;
int cost[1010][1010];
int used[1010];
int tata[1010];
int nr,n,m;
struct bla
{
    int a,b;
} date[10101];
void read ()
{ int x,y,c;
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        cin>>x>>y>>c;
        t[x].push_back(y);
        t[y].push_back(x);
        cost[x][y]=c;
        cost[y][x]=c;
        date[i].a=x; date[i].b=y;
    }
}
bool bfs ()
{
    ++nr;
    coada.push(1);
    used[1]=nr;
    while(!coada.empty())
    {
        int nod=coada.front();
        coada.pop();
        for(vector <int> :: iterator i=t[nod].begin(); i!=t[nod].end(); ++i)
        {
            if(used[*i]<nr && cost[nod][*i]>0)
            {
                used[*i]=nr;
                coada.push(*i);
                tata[*i]=nod;
            }
        }
    }
    if(used[n]==nr) return true;
    return false;
}
void fflux ()
{
    while(bfs())
    {
        for(vector <int> :: iterator i=t[n].begin(); i!=t[n].end(); ++i )
        {
            if(used[*i]==nr && cost[*i][n]>0)
            {
                int minim=INT_MAX;
                tata[n]=*i;
                for(int x=n;x!=1;x=tata[x]) minim=min(cost[tata[x]][x],minim);
                for(int x=n;x!=1;x=tata[x]) cost[tata[x]][x]-=minim,cost[x][tata[x]]-=minim;
            }
        }
    }
}
void bfs2 ()
{
    coada.push(n);
    tata[n]=-1;
    while(!coada.empty())
    {
        int nod=coada.front();
        coada.pop();
        for(vector <int> :: iterator i=t[nod].begin(); i!=t[nod].end(); ++i)
            if(tata[*i]!=-1 && cost[nod][*i]>0)
            {
                coada.push(*i);
                tata[*i]=-1;
            }
    }
}
void solve ()
{ int p=0;
    for(int i=1;i<=m;i++)
    {
        if(cost[date[i].a][date[i].b]==0 && ((used[date[i].a]==nr && tata[date[i].b]==-1) || (used[date[i].b]==nr && tata[date[i].a]==-1)))
            ++p;
    }
    cout<<p<<"\n";
    for(int i=1;i<=m;i++)
    {
        if(cost[date[i].a][date[i].b]==0 && ((used[date[i].a]==nr && tata[date[i].b]==-1) || (used[date[i].b]==nr && tata[date[i].a]==-1)))
            cout<<i<<"\n";
    }
}
int main()
{
    read();
    fflux();
    bfs2();
    solve();
    cin.close();
    cout.close();
    return 0;
}