Cod sursa(job #917858)

Utilizator valentina506Moraru Valentina valentina506 Data 18 martie 2013 13:33:09
Problema Critice Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.14 kb
#include<fstream>
#include<vector>
#include<queue>
#include<cstring>
#include<algorithm>
#define inf 1<<30
using namespace std;
int n,m,i,j,C[1001][1001],t[1001],z,nod,fluxmax,fluxct,nr;
vector<int>a[1001],b;
queue<int>q;
bool uz[1001],cr[10001];
struct muchie
{
    int x,y;
};
muchie M[10001];
int bf()
{
    int i,x;
    memset(uz,0,sizeof(uz));
    q.push(1);
    uz[1]=1;
    while(!q.empty())
    {
        x=q.front();
        if(x!=n)
        {
            for(i=0;i<a[x].size();++i)
            if(!uz[a[x][i]]&&C[x][a[x][i]])
            {
                uz[a[x][i]]=1;
                t[a[x][i]]=x;
                q.push(a[x][i]);
            }
        }
        q.pop();
    }
    return uz[n];
}
int bf2(int x,int y)
{
    int i;
    memset(uz,0,sizeof(uz));
    q.push(x);
    q.push(y);
    uz[x]=uz[y]=1;
    while(!q.empty())
    {
        x=q.front();
        if(x!=1&&x!=n)
        for(i=0;i<a[x].size();++i)
        if(!uz[a[x][i]]&&C[x][a[x][i]])
        {
            uz[a[x][i]]=1;
            q.push(a[x][i]);
        }
        q.pop();
    }
    if(uz[1]&&uz[n])
    return 1;
    return 0;

}
int main()
{
    ifstream f("critice.in");
    ofstream g("critice.out");
    f>>n>>m;
    for(i=1;i<=m;++i)
    {
        f>>M[i].x>>M[i].y>>z;
        a[M[i].x].push_back(M[i].y);
        a[M[i].y].push_back(M[i].x);
        C[M[i].x][M[i].y] = C[M[i].y][M[i].x] =z;
    }
    fluxmax=0;
    while(bf())
    {
        for(i=0;i<a[n].size();++i)
        if(uz[a[n][i]]&&C[ a[n][i] ][n]>0 )
        {
            t[n]=a[n][i];
            fluxct=inf;
            for(nod=n;nod!=1;nod=t[nod])
            fluxct=min(fluxct,C[t[nod]][nod]);

            if(fluxct)
            {

                for(nod=n;nod!=1;nod=t[nod])
                {
                    C[t[nod]][nod]-=fluxct;
                    C[nod][t[nod]]-=fluxct;
                }
                fluxmax+=fluxct;
            }
        }
    }

    for(i=1;i<=m;++i)
    if(C[M[i].x][M[i].y]==0&&bf2(M[i].x,M[i].y))
    ++nr,b.push_back(i);

    g<<nr<<'\n';
    for(i=0;i<nr;++i)
    g<<b[i]<<'\n';
}