Cod sursa(job #1245681)

Utilizator gapdanPopescu George gapdan Data 19 octombrie 2014 20:30:17
Problema Critice Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.29 kb
#include<fstream>
#include<queue>
#include<cstring>
#define min(a,b) (a>b?b:a)
using namespace std;
struct muchie
{
    int x,y;
}muc[1001];
vector<int>v[1001];
int f[1001][1001],c[1001][1001],tata[1001];
int n,m,x,y,z,viz[1001],viz1[1001],sol[1001],k,i;
int q[1001],st,dr;
int BFS()
{
   q[1]=1;st=1;dr=1;
    memset(viz,0,sizeof(viz));
    viz[1]=1;
    while(st<=dr)
    {
        int nod=q[st];
        st++;
        for (i=0;i<v[nod].size();++i)
        {
            int varf=v[nod][i];
            if (c[nod][varf]>f[nod][varf] && viz[varf]==0)
            {
                q[++dr]=varf;
                tata[varf]=nod;
                viz[varf]=1;
            }
        }
    }
    return viz[n];
}
void BFS1()
{
    q[1]=n;st=1,dr=1;
    viz1[n]=1;
    while(st<=dr)
    {
        int nod=q[st];
        ++st;
        for (i=0;i<v[nod].size();++i)
        {
            int varf=v[nod][i];
            if (c[nod][varf]>f[nod][varf]*(-1) && viz1[varf]==0)
            {
                q[++dr]=varf;
                tata[varf]=nod;
                viz1[varf]=1;
            }
        }
    }
}
int main()
{
    ifstream cin("critice.in");
    ofstream cout("critice.out");
    cin>>n>>m;
    for (i=1;i<=m;++i)
    {
        cin>>x>>y>>z;
        muc[i].x=x;
        muc[i].y=y;
        v[x].push_back(y);
        v[y].push_back(x);
        c[x][y]=z;
        c[y][x]=z;
    }
    int flux=0;
    while(BFS())
    {
        for (i=0;i<v[n].size();++i)
        {
            int nod=v[n][i];
            if (f[nod][n]==c[nod][n] || !viz[nod]) continue;
            tata[n]=nod;
            int fmin=c[tata[nod]][nod]-f[tata[nod]][nod];
            for (nod=n;nod!=1;nod=tata[nod])
                fmin=min(fmin,c[tata[nod]][nod]-f[tata[nod]][nod]);
            if (fmin==0) continue;
            for (nod=n;nod!=1;nod=tata[nod])
                {
                    f[tata[nod]][nod]+=fmin;
                    f[nod][tata[nod]]-=fmin;
                }
            flux+=fmin;
        }
    }
    x=BFS();
    BFS1();
    for (i=1;i<=m;++i)
        if ((viz[muc[i].x]==1 && viz1[muc[i].y]==1) || (viz[muc[i].y]==1 && viz1[muc[i].x]==1))
            sol[++k]=i;
    cout<<k<<"\n";
    for (i=1;i<=k;++i) cout<<sol[i]<<"\n";
    return 0;
}