Pagini recente » Cod sursa (job #2938378) | Cod sursa (job #1377032) | Cod sursa (job #2551953) | Cod sursa (job #1016472) | Cod sursa (job #2246102)
#include<fstream>
#include<bitset>
#include<queue>
#define DIM 1001
#define inf 2000000000
using namespace std;
ifstream fin("critice.in");
ofstream fout("critice.out");
int n,m,i,mini,ve,a,b,cap, sol[DIM],nod,j,va[10001],vb[10001],nr;
int c[DIM][DIM], f[DIM][DIM];
int t[DIM];
bitset<DIM> d;
deque<int> z;
vector<int> v[DIM];
int bfs()
{
int i, nod,vecin;
d.reset();
z.push_back(1);
d[1]=1;
while(!z.empty())
{
nod=z[0];z.pop_front();
for(i=0;i<v[nod].size();i++)
{
vecin=v[nod][i];
if(c[nod][vecin]>f[nod][vecin]&&d[vecin]==0)
{
z.push_back(vecin);
d[vecin]=1;t[vecin]=nod;
}
}
}
return d[n];
}
void solve(int start, int val)
{
int i, nod,vecin;
d.reset();
z.push_back(start);
d[start]=1;
while(!z.empty())
{
nod=z[0];z.pop_front();
for(i=0;i<v[nod].size();i++)
{
vecin=v[nod][i];
if(!((f[vecin][nod]==c[vecin][nod])||(f[nod][vecin]==c[nod][vecin]))&&d[vecin]==0)
{
z.push_back(vecin);
d[vecin]=1;t[vecin]=nod;
}
}
}
for(i=1;i<=n;i++)
if(d[i])
sol[i]+=val;
}
int main()
{
fin>>n>>m;
for(i=1;i<=m;i++)
{
fin>>a>>b>>cap;
va[i]=a;vb[i]=b;
v[a].push_back(b);
v[b].push_back(a);
c[a][b]=cap;
c[b][a]=cap;
}
while(bfs()==1)
{
for(i=0;i<v[n].size();i++)
{
ve=v[n][i];
if(d[ve]==1&&c[ve][n]>f[ve][n])
{
a=ve;b=n;
mini=c[ve][n]-f[ve][n];
while(a!=0)
{
mini=min(mini, c[a][b]-f[a][b]);
b=a;a=t[a];
}
a=ve;b=n;
while(a!=0)
{
f[a][b]+=mini;
f[b][a]-=mini;
b=a;a=t[a];
}
f[n][ve]+=mini;
}
}
}
solve(1, 1);solve(n, 2);
for(int x=1;x<=m;x++)
{
i=va[x];j=vb[x];
if(((f[i][j]==c[i][j])||(f[j][i]==c[j][i]))&&((sol[i]==1&&sol[j]==2)||(sol[i]==2&&sol[j]==1)))
nr++;
}
fout<<nr<<"\n";
for(int x=1;x<=m;x++)
{
i=va[x];j=vb[x];
if(((f[i][j]==c[i][j])||(f[j][i]==c[j][i]))&&((sol[i]==1&&sol[j]==2)||(sol[i]==2&&sol[j]==1)))
fout<<x<<"\n";
}
return 0;
}