alpar@921: #ifndef LEMON_PREFLOW_PUSH_HH alpar@921: #define LEMON_PREFLOW_PUSH_HH athos@331: athos@331: //#include athos@331: #include athos@331: #include athos@331: #include athos@331: //#include "pf_hiba.hh" athos@331: //#include athos@331: //#include athos@331: #include athos@331: //#include athos@331: athos@331: using namespace std; athos@331: alpar@921: namespace lemon { athos@331: athos@331: template athos@331: class preflow_push { athos@331: athos@331: //Useful typedefs athos@331: typedef typename Graph::Node Node; athos@331: typedef typename Graph::NodeIt NodeIt; athos@331: typedef typename Graph::Edge Edge; athos@331: typedef typename Graph::OutEdgeIt OutEdgeIt; athos@331: typedef typename Graph::InEdgeIt InEdgeIt; athos@331: athos@331: athos@331: //--------------------------------------------- athos@331: //Parameters of the algorithm athos@331: //--------------------------------------------- athos@331: //Fully examine an active node until excess becomes 0 athos@331: enum node_examination_t {examine_full, examine_to_relabel}; athos@331: //No more implemented yet:, examine_only_one_edge}; athos@331: node_examination_t node_examination; athos@331: //Which implementation to be used athos@331: enum implementation_t {impl_fifo, impl_highest_label}; athos@331: //No more implemented yet:}; athos@331: implementation_t implementation; athos@331: //--------------------------------------------- athos@331: //Parameters of the algorithm athos@331: //--------------------------------------------- athos@331: athos@331: private: athos@331: //input athos@331: Graph& G; athos@331: Node s; athos@331: Node t; athos@331: typename Graph::EdgeMap &capacity; athos@331: athos@331: //output athos@331: typename Graph::EdgeMap preflow; athos@331: T maxflow_value; athos@331: athos@331: //auxiliary variables for computation athos@331: //The number of the nodes athos@331: int number_of_nodes; athos@331: //A nodemap for the level athos@331: typename Graph::NodeMap level; athos@331: //A nodemap for the excess athos@331: typename Graph::NodeMap excess; athos@331: athos@331: //Number of nodes on each level athos@331: vector num_of_nodes_on_level; athos@331: athos@331: //For the FIFO implementation athos@331: list fifo_nodes; athos@331: //For 'highest label' implementation athos@331: int highest_active; athos@331: //int second_highest_active; athos@331: vector< list > active_nodes; athos@331: athos@331: public: athos@331: athos@331: //Constructing the object using the graph, source, sink and capacity vector athos@331: preflow_push( athos@331: Graph& _G, athos@331: Node _s, athos@331: Node _t, athos@331: typename Graph::EdgeMap & _capacity) athos@331: : G(_G), s(_s), t(_t), athos@331: capacity(_capacity), athos@331: preflow(_G), athos@331: //Counting the number of nodes athos@331: //number_of_nodes(count(G.first())), athos@331: number_of_nodes(G.nodeNum()), athos@331: athos@331: level(_G), athos@331: excess(_G)//, athos@331: // Default constructor: active_nodes() athos@331: { athos@331: //Simplest parameter settings athos@331: node_examination = examine_full;//examine_to_relabel;// athos@331: //Which implementation to be usedexamine_full athos@331: implementation = impl_highest_label;//impl_fifo; athos@331: athos@331: // athos@331: num_of_nodes_on_level.resize(2*number_of_nodes-1); athos@331: num_of_nodes_on_level.clear(); athos@331: athos@331: switch(implementation){ athos@331: case impl_highest_label :{ athos@331: active_nodes.clear(); athos@331: active_nodes.resize(2*number_of_nodes-1); athos@331: athos@331: break; athos@331: } athos@331: default: athos@331: break; athos@331: } athos@331: athos@331: } athos@331: athos@331: //Returns the value of a maximal flow athos@331: T run(); athos@331: athos@331: typename Graph::EdgeMap getmaxflow(){ athos@331: return preflow; athos@331: } athos@331: athos@331: athos@331: private: athos@331: //For testing purposes only athos@331: //Lists the node_properties athos@331: void write_property_vector(typename Graph::NodeMap a, athos@331: //node_property_vector a, athos@331: char* prop_name="property"){ athos@331: for(NodeIt i=G.template first(); G.valid(i); G.next(i)) { athos@331: cout<<"Node id.: "<=0 && active_nodes[highest_active].empty() ){ athos@331: --highest_active; athos@331: } athos@331: athos@331: if( highest_active>=0) { athos@331: athos@331: athos@331: Node a=active_nodes[highest_active].front(); athos@331: active_nodes[highest_active].pop_front(); athos@331: athos@331: return a; athos@331: } athos@331: else { athos@331: return INVALID; athos@331: } athos@331: athos@331: break; athos@331: athos@331: } athos@331: case impl_fifo : { athos@331: athos@331: if( ! fifo_nodes.empty() ) { athos@331: Node a=fifo_nodes.front(); athos@331: fifo_nodes.pop_front(); athos@331: return a; athos@331: } athos@331: else { athos@331: return INVALID; athos@331: } athos@331: break; athos@331: } athos@331: } athos@331: // athos@331: return INVALID; athos@331: } athos@331: athos@331: //Puts node 'a' among the active nodes athos@331: void make_active(const Node& a){ athos@331: //s and t never become active athos@331: if (a!=s && a!= t){ athos@331: switch(implementation){ athos@331: case impl_highest_label : athos@331: active_nodes[level[a]].push_back(a); athos@331: break; athos@331: case impl_fifo : athos@331: fifo_nodes.push_back(a); athos@331: break; athos@331: } athos@331: athos@331: } athos@331: athos@331: //Update highest_active label athos@331: if (highest_active(); G.valid(i); G.next(i)) { athos@331: level.set(i,0); athos@331: excess.set(i,0); athos@331: for(OutEdgeIt j=G.template first(i); G.valid(j); G.next(j)) athos@331: preflow.set(j, 0); athos@331: } athos@331: num_of_nodes_on_level[0]=number_of_nodes; athos@331: highest_active=0; athos@331: //--------------------------------------- athos@331: //Initialize parameters athos@331: //--------------------------------------- athos@331: athos@331: athos@331: //------------------------------------ athos@331: //This is the only part that uses BFS athos@331: //------------------------------------ athos@331: athos@331: /*Reverse_bfs from t, to find the starting level.*/ athos@331: //Copyright: Jacint athos@331: change_level_to(t,0); athos@331: athos@331: std::queue bfs_queue; athos@331: bfs_queue.push(t); athos@331: athos@331: while (!bfs_queue.empty()) { athos@331: athos@331: Node v=bfs_queue.front(); athos@331: bfs_queue.pop(); athos@331: int l=level[v]+1; athos@331: athos@331: InEdgeIt e; athos@331: for(G.first(e,v); G.valid(e); G.next(e)) { alpar@986: Node w=G.source(e); athos@331: if ( level[w] == number_of_nodes && w != s ) { athos@331: bfs_queue.push(w); athos@331: //Node first=level_list[l]; athos@331: //if ( G.valid(first) ) left.set(first,w); athos@331: //right.set(w,first); athos@331: //level_list[l]=w; athos@331: change_level_to(w, l); athos@331: //level.set(w, l); athos@331: } athos@331: } athos@331: } athos@331: change_level_to(s,number_of_nodes); athos@331: //level.set(s,number_of_nodes); athos@331: athos@331: /* athos@331: //Setting starting level values using reverse bfs athos@331: reverse_bfs rev_bfs(G,t); athos@331: rev_bfs.run(); athos@331: //write_property_vector(rev_bfs.dist,"rev_bfs"); athos@331: for(NodeIt i=G.template first(); G.valid(i); G.next(i)) { athos@331: change_level_to(i,rev_bfs.dist(i)); athos@331: //level.put(i,rev_bfs.dist.get(i)); athos@331: } athos@331: */ athos@331: //------------------------------------ athos@331: //This is the only part that uses BFS athos@331: //------------------------------------ athos@331: athos@331: athos@331: //Starting level of s athos@331: change_level_to(s,number_of_nodes); athos@331: //level.put(s,number_of_nodes); athos@331: athos@331: athos@331: //we push as much preflow from s as possible to start with athos@331: for(OutEdgeIt j=G.template first(s); G.valid(j); G.next(j)){ athos@331: modify_preflow(j,capacity[j] ); alpar@986: make_active(G.target(j)); alpar@986: int lev=level[G.target(j)]; athos@331: if(highest_activepreflow[j]){ alpar@986: if(level[G.source(j)]==level[G.target(j)]+1){ athos@331: return true; athos@331: } athos@331: else{ alpar@986: if (level[G.target(j)] < new_level) alpar@986: new_level=level[G.target(j)]; athos@331: } athos@331: } athos@331: return false; athos@331: } athos@331: athos@331: //If the preflow is greater than 0 on the given edge athos@331: //then the edge reversd is an edge in the residual graph athos@331: bool is_admissible_backward_edge(Edge j, int& new_level){ athos@331: athos@331: if (0 athos@331: T preflow_push::run() { athos@331: athos@331: preprocess(); athos@331: //write_property_vector(level,"level"); athos@331: T e,v; athos@331: Node a; athos@331: while (a=get_active_node(), G.valid(a)){ athos@331: athos@331: //cout<(a); athos@331: while (G.valid(j) && e){ athos@331: athos@331: if (is_admissible_forward_edge(j,new_level)){ athos@331: v=min(e,capacity[j] - preflow[j]); athos@331: e -= v; athos@331: //New node might become active alpar@986: if (excess[G.target(j)]==0){ alpar@986: make_active(G.target(j)); athos@331: } athos@331: modify_preflow(j,v); athos@331: } athos@331: G.next(j); athos@331: } athos@331: } athos@331: //In edges to node a athos@331: { athos@331: InEdgeIt j=G.template first(a); athos@331: while (G.valid(j) && e){ athos@331: if (is_admissible_backward_edge(j,new_level)){ athos@331: v=min(e,preflow[j]); athos@331: e -= v; athos@331: //New node might become active alpar@986: if (excess[G.source(j)]==0){ alpar@986: make_active(G.source(j)); athos@331: } athos@331: modify_preflow(j,-v); athos@331: } athos@331: G.next(j); athos@331: } athos@331: } athos@331: athos@331: //if (G.id(a)==999) athos@331: //cout<