Pagini recente » Cod sursa (job #1338543) | Cod sursa (job #193693) | Borderou de evaluare (job #1036648) | Cod sursa (job #1868564) | Cod sursa (job #1817745)
#include <iostream>
#include<fstream>
#include<vector>
#include<algorithm>
#define inf 1000000
using namespace std;
ifstream f("critice.in");
ofstream g("critice.out");
int sz,c[1000000],C[1001][1001],F[1001][1001],k,i,flow,fmin,n,m,father[1001],x,y,cst,start,finish,vz[1001],nod2,nod1;
vector<int>v[1001],sol;
bool viz[1001],taken[10001];
struct muchie
{
int x,y;
};
muchie mu[10001];
int bfs()
{
sz=1;
c[1]=1;
int i,nod;
for(i=1; i<=n; i++)
viz[i]=0;
viz[1]=1;
for(i=1; i<=sz; i++)
{
nod=c[i];
if(nod==n)
continue;
for(k=0; k<v[nod].size(); k++)
{
int vertex=v[nod][k];
if(viz[vertex] || F[nod][vertex]==0)
continue;
c[++sz]=vertex;
viz[vertex]=1;
father[vertex]=nod;
}
}
return viz[n];
}
void cauta(int a,int b)
{
bool ok=1;
for(int i=1; i<=m && ok; i++)
if(mu[i].x==a && mu[i].y==b || (mu[i].x==b && mu[i].y==a))
{
ok=0;
if(!taken[i])
sol.push_back(i),taken[i]=1;
}
}
void df(int nod,int cnt)
{
if(cnt<=1)
{
viz[nod]=1;
if(nod==n)
cauta(nod1,nod2);
for(int k=0; k<v[nod].size(); k++)
{
if(!viz[v[nod][k]])
{
if(F[nod][v[nod][k]]==0)
{
if(cnt<1)
{
nod1=nod;
nod2=v[nod][k];
df(v[nod][k],cnt+1);
}
}
else
df(v[nod][k],cnt);
}
}
viz[nod]=0;
}
else
nod1=nod2=0;
}
int main()
{
f>>n>>m;
for(i=1; i<=m; i++)
{
f>>x>>y>>cst;
v[x].push_back(y);
v[y].push_back(x);
F[x][y]=cst;
mu[i].x=x;
mu[i].y=y;
}
while(bfs())
for(k=0; k<v[n].size(); k++)
{
int nod=v[n][k];
if(F[nod][n]==0 || !viz[nod])
continue;
father[n]=nod;
fmin=inf;
for(i=n; i!=1; i=father[i])
fmin=min(fmin,F[father[i]][i]);
for(i=n; i!=1; i=father[i])
{
F[father[i]][i]-=fmin;
F[i][father[i]]+=fmin;
}
flow+=fmin;
}
for(i=1; i<=n; i++)
viz[i]=0;
df(1,0);
sort(sol.begin(),sol.end());
for(i=0; i<sol.size(); i++)
g<<sol[i]<<'\n';
}