Pagini recente » Cod sursa (job #457905) | Cod sursa (job #546006) | Cod sursa (job #380560) | Cod sursa (job #2607653) | Cod sursa (job #1402172)
#include <fstream>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;
ifstream f("critice.in");
ofstream g("critice.out");
int N,M,MaxFlow;
const int NMax = 1005;
const int MMax = 10005;
const int INF = 0x3f3f3f3f;
vector <int> G[NMax];
queue <int> Q;
int C[NMax][NMax],Use[NMax],F[NMax][NMax],TT[NMax];
pair <int,int> E[MMax];
vector <int> Sol;
void Read()
{
f>>N>>M;
for(int i=1; i<= M ; i++)
{
int x,y,c;
f>>x>>y>>c;
G[x].push_back(y);
G[y].push_back(x);
C[x][y] += c;
C[y][x] += c;
E[i]=make_pair(x,y);
}
}
int BFS()
{
for(int i=1;i<=N;i++)
Use[i]=0;
Q.push(1); Use[1] = 1;
while(!Q.empty())
{
int Node = Q. front(); Q. pop();
if(Node == N) continue;
for(vector <int> :: iterator it=G[Node].begin(); it!=G[Node].end(); it++)
{
int Neighbour = *it;
if(!Use[Neighbour] && C[Node][Neighbour]-F[Node][Neighbour]>0)
{
TT[Neighbour] = Node;
Q.push(Neighbour);
Use[Neighbour] = 1;
}
}
}
return Use[N];
}
void maxFlow()
{
while(BFS())
{
for(vector <int> :: iterator it=G[N].begin(); it!=G[N].end(); it++)
{
int Vecin = *it;
if(!Use[Vecin] || C[Vecin][N]-F[Vecin][N] == 0) continue;
TT[N] = Vecin;
int Fmin = INF;
for(int i = N ; i != 1 ; i = TT[i])
Fmin = min(Fmin,C[TT[i]][i]-F[TT[i]][i]);
MaxFlow += Fmin;
for(int i = N ; i != 1 ; i = TT[i])
{
F[TT[i]][i] += Fmin;
F[i][TT[i]] -= Fmin;
}
}
}
}
bool BFS1(int x)
{
Q.push(1);
for(int i=1;i<=N;i++)
Use[i]=0;
Use[1]=1;
while(!Q.empty())
{
int node=Q.front();
Q.pop();
if(node==x)
{
while(!Q.empty())
Q.pop();
return 1;
}
for(vector <int> :: iterator it=G[node].begin(); it!=G[node].end(); it++)
{
int neighb=*it;
if(Use[neighb]==0 && C[node][neighb]-F[node][neighb]>0)
{
Q.push(neighb);
Use[neighb]=1;
}
}
}
return 0;
}
bool BFS2(int x)
{
Q.push(x);
for(int i=1;i<=N;i++)
Use[i]=0;
Use[x]=1;
while(!Q.empty())
{
int node=Q.front();
Q.pop();
if(node==N)
{
while(!Q.empty())
Q.pop();
return 1;
}
for(vector <int> :: iterator it=G[node].begin(); it!=G[node].end(); it++)
{
int neighb=*it;
if(Use[neighb]==0 && C[node][neighb]-F[node][neighb]>0)
{
Q.push(neighb);
Use[neighb]=1;
}
}
}
return 0;
}
void findEdges()
{
for(int i=1;i<=M;i++)
{
int x=E[i].first,y=E[i].second;
if(F[x][y]<0)
swap(x,y);
if(BFS1(x) && BFS2(y))
Sol.push_back(i);
}
}
void Print()
{
g<<Sol.size()<<"\n";
for(int i=0;i<Sol.size();i++)
g<<Sol[i]<<"\n";
}
int main()
{
Read();
maxFlow();
findEdges();
Print();
return 0;
}