#include <string>
#include <iostream>
#include <map>
#include <vector>

#include <kruskal.h>
#include <list_graph.h>


using namespace std;
using namespace hugo;

class string_int_map : public map<string,int> {
public:
  int get(const string &s) {
    // Bocs, ez igy gaaaany, de nem volt kedvem utananezni, hogy
    // hogy is mukodik ez a map :)
    if( count(s) == 0 ) {
      operator[](s) = -1;
    }
    return operator[](s);
  }
  void set(const string &s, int i) {
      operator[](s) = i;
  }
};


// Egy olyan "map", ami nem tud semmit, csak a typedef-eket.
// Valami elegansabb megoldas kene a Kruskalban...

template <typename K, typename V>
class DummyMap {
public:
  typedef K KeyType;
  typedef V ValueType;
};

int main() {

  typedef ListGraph::Node Node;
  typedef ListGraph::Edge Edge;
  typedef ListGraph::NodeIt NodeIt;
  typedef ListGraph::EdgeIt EdgeIt;

  ListGraph G;

  Node s=G.addNode();
  Node v1=G.addNode();
  Node v2=G.addNode();
  Node v3=G.addNode();
  Node v4=G.addNode();
  Node t=G.addNode();
  
  Edge e1 = G.addEdge(s, v1);
  Edge e2 = G.addEdge(s, v2);
  Edge e3 = G.addEdge(v1, v2);
  Edge e4 = G.addEdge(v2, v1);
  Edge e5 = G.addEdge(v1, v3);
  Edge e6 = G.addEdge(v3, v2);
  Edge e7 = G.addEdge(v2, v4);
  Edge e8 = G.addEdge(v4, v3);
  Edge e9 = G.addEdge(v3, t);
  Edge e10 = G.addEdge(v4, t);

  typedef ListGraph::EdgeMap<double> ECostMap;
  typedef ListGraph::EdgeMap<bool> EBoolMap;

  ECostMap edge_cost_map(G, 2);
  EBoolMap tree_map(G);
  

  cout << "Uniform 2-es koltseggel: " 
       << kruskal1(G, edge_cost_map, tree_map)
       << endl;


  edge_cost_map.set(e1, -10);
  edge_cost_map.set(e2, -9);
  edge_cost_map.set(e3, -8);
  edge_cost_map.set(e4, -7);
  edge_cost_map.set(e5, -6);
  edge_cost_map.set(e6, -5);
  edge_cost_map.set(e7, -4);
  edge_cost_map.set(e8, -3);
  edge_cost_map.set(e9, -2);
  edge_cost_map.set(e10, -1);

  vector<Edge> tree_edge_vec;

  cout << "Nemkonst koltseggel (-31): "
       << kruskal2(G, edge_cost_map, back_inserter(tree_edge_vec))
       << endl;

  int i = 1;
  for(vector<Edge>::iterator e = tree_edge_vec.begin();
      e != tree_edge_vec.end(); ++e, ++i) {
    cout << i << ". el: " << *e << endl;
  }


//   typedef MinCostTreeKruskal<ListGraph, ECostMap, EBoolMap> MCTK;

//   MCTK mctk(G, edge_cost_map, tree_map);
//   double k0lts = mctk.run();

//   cout << "Uniform 2-es koltseggel: " << k0lts << endl;

//   // Max koltsegu fa szamitasa elore megrendezett koltseg vektorbol:
//   typedef MinCostTreeKruskal<ListGraph, DummyMap<Edge,int>, EBoolMap> MCTK2;
//   MCTK2 mctk2(G, DummyMap<Edge,int>(), tree_map);
//   MCTK2::EdgeCostVector ecv;
//   ecv.push_back(make_pair(e1, 10));
//   ecv.push_back(make_pair(e2, 9));
//   ecv.push_back(make_pair(e3, 8));
//   ecv.push_back(make_pair(e4, 7));
//   ecv.push_back(make_pair(e5, 6));
//   ecv.push_back(make_pair(e6, 5));
//   ecv.push_back(make_pair(e7, 4));
//   ecv.push_back(make_pair(e8, 3));
//   ecv.push_back(make_pair(e9, 2));
//   ecv.push_back(make_pair(e10, 1));

//   k0lts = mctk2.run(ecv);
//   cout << "Max koltsegu fa elore megrendezett koltseg vektorbol: 31 = "
//        << k0lts << endl;


  return 0;
}