beckerjc@150: #include <string>
beckerjc@150: #include <iostream>
beckerjc@150: #include <map>
beckerjc@150: 
beckerjc@150: #include <kruskal.h>
beckerjc@218: #include <list_graph.h>
beckerjc@150: 
beckerjc@150: 
beckerjc@150: using namespace std;
beckerjc@150: using namespace hugo;
beckerjc@150: 
beckerjc@150: class string_int_map : public map<string,int> {
beckerjc@150: public:
beckerjc@150:   int get(const string &s) {
beckerjc@150:     // Bocs, ez igy gaaaany, de nem volt kedvem utananezni, hogy
beckerjc@150:     // hogy is mukodik ez a map :)
beckerjc@150:     if( count(s) == 0 ) {
beckerjc@150:       operator[](s) = -1;
beckerjc@150:     }
beckerjc@150:     return operator[](s);
beckerjc@150:   }
beckerjc@150:   void set(const string &s, int i) {
beckerjc@150:       operator[](s) = i;
beckerjc@150:   }
beckerjc@150: };
beckerjc@150: 
beckerjc@150: 
beckerjc@218: // Egy olyan "map", ami nem tud semmit, csak a typedef-eket.
beckerjc@218: // Valami elegansabb megoldas kene a Kruskalban...
beckerjc@218: 
beckerjc@218: template <typename K, typename V>
beckerjc@218: class DummyMap {
beckerjc@218: public:
beckerjc@218:   typedef K KeyType;
beckerjc@218:   typedef V ValueType;
beckerjc@218: };
beckerjc@218: 
beckerjc@150: int main() {
beckerjc@150: 
beckerjc@218:   typedef ListGraph::Node Node;
beckerjc@218:   typedef ListGraph::Edge Edge;
beckerjc@150:   typedef ListGraph::NodeIt NodeIt;
beckerjc@150:   typedef ListGraph::EdgeIt EdgeIt;
beckerjc@150: 
beckerjc@150:   ListGraph G;
beckerjc@150: 
beckerjc@218:   Node s=G.addNode();
beckerjc@218:   Node v1=G.addNode();
beckerjc@218:   Node v2=G.addNode();
beckerjc@218:   Node v3=G.addNode();
beckerjc@218:   Node v4=G.addNode();
beckerjc@218:   Node t=G.addNode();
beckerjc@150:   
beckerjc@218:   Edge e1 = G.addEdge(s, v1);
beckerjc@218:   Edge e2 = G.addEdge(s, v2);
beckerjc@218:   Edge e3 = G.addEdge(v1, v2);
beckerjc@218:   Edge e4 = G.addEdge(v2, v1);
beckerjc@218:   Edge e5 = G.addEdge(v1, v3);
beckerjc@218:   Edge e6 = G.addEdge(v3, v2);
beckerjc@218:   Edge e7 = G.addEdge(v2, v4);
beckerjc@218:   Edge e8 = G.addEdge(v4, v3);
beckerjc@218:   Edge e9 = G.addEdge(v3, t);
beckerjc@218:   Edge e10 = G.addEdge(v4, t);
beckerjc@150: 
beckerjc@218:   typedef ListGraph::EdgeMap<double> ECostMap;
beckerjc@218:   typedef ListGraph::EdgeMap<bool> EBoolMap;
beckerjc@218: 
beckerjc@218:   ECostMap edge_cost_map(G, 2);
beckerjc@218:   EBoolMap tree_map(G);
beckerjc@150:   
beckerjc@218:   typedef MinCostTreeKruskal<ListGraph, ECostMap, EBoolMap> MCTK;
beckerjc@150: 
beckerjc@218:   MCTK mctk(G, edge_cost_map, tree_map);
beckerjc@218:   double k0lts = mctk.run();
beckerjc@218: 
beckerjc@218:   cout << "Uniform 2-es koltseggel: " << k0lts << endl;
beckerjc@218: 
beckerjc@218:   // Max koltsegu fa szamitasa elore megrendezett koltseg vektorbol:
beckerjc@218:   typedef MinCostTreeKruskal<ListGraph, DummyMap<Edge,int>, EBoolMap> MCTK2;
beckerjc@218:   MCTK2 mctk2(G, DummyMap<Edge,int>(), tree_map);
beckerjc@218:   MCTK2::EdgeCostVector ecv;
beckerjc@218:   ecv.push_back(make_pair(e1, 10));
beckerjc@218:   ecv.push_back(make_pair(e2, 9));
beckerjc@218:   ecv.push_back(make_pair(e3, 8));
beckerjc@218:   ecv.push_back(make_pair(e4, 7));
beckerjc@218:   ecv.push_back(make_pair(e5, 6));
beckerjc@218:   ecv.push_back(make_pair(e6, 5));
beckerjc@218:   ecv.push_back(make_pair(e7, 4));
beckerjc@218:   ecv.push_back(make_pair(e8, 3));
beckerjc@218:   ecv.push_back(make_pair(e9, 2));
beckerjc@218:   ecv.push_back(make_pair(e10, 1));
beckerjc@218: 
beckerjc@218:   k0lts = mctk2.run(ecv);
beckerjc@218:   cout << "Max koltsegu fa elore megrendezett koltseg vektorbol: 31 = "
beckerjc@218:        << k0lts << endl;
beckerjc@218: 
beckerjc@150: 
beckerjc@150:   return 0;
beckerjc@150: }