bucket_sort.hpp 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144
  1. //=======================================================================
  2. // Copyright 2007 Aaron Windsor
  3. //
  4. // Distributed under the Boost Software License, Version 1.0. (See
  5. // accompanying file LICENSE_1_0.txt or copy at
  6. // http://www.boost.org/LICENSE_1_0.txt)
  7. //=======================================================================
  8. #ifndef __BUCKET_SORT_HPP__
  9. #define __BUCKET_SORT_HPP__
  10. #include <vector>
  11. #include <algorithm>
  12. #include <boost/property_map/property_map.hpp>
  13. namespace boost
  14. {
  15. template <typename ItemToRankMap>
  16. struct rank_comparison
  17. {
  18. rank_comparison(ItemToRankMap arg_itrm) : itrm(arg_itrm) {}
  19. template <typename Item>
  20. bool operator() (Item x, Item y) const
  21. {
  22. return get(itrm, x) < get(itrm, y);
  23. }
  24. private:
  25. ItemToRankMap itrm;
  26. };
  27. template <typename TupleType,
  28. int N,
  29. typename PropertyMapWrapper = identity_property_map>
  30. struct property_map_tuple_adaptor :
  31. public put_get_helper< typename PropertyMapWrapper::value_type,
  32. property_map_tuple_adaptor
  33. <TupleType, N, PropertyMapWrapper>
  34. >
  35. {
  36. typedef typename PropertyMapWrapper::reference reference;
  37. typedef typename PropertyMapWrapper::value_type value_type;
  38. typedef TupleType key_type;
  39. typedef readable_property_map_tag category;
  40. property_map_tuple_adaptor() {}
  41. property_map_tuple_adaptor(PropertyMapWrapper wrapper_map) :
  42. m_wrapper_map(wrapper_map)
  43. {}
  44. inline value_type operator[](const key_type& x) const
  45. {
  46. return get(m_wrapper_map, get<n>(x));
  47. }
  48. static const int n = N;
  49. PropertyMapWrapper m_wrapper_map;
  50. };
  51. // This function sorts a sequence of n items by their ranks in linear time,
  52. // given that all ranks are in the range [0, range). This sort is stable.
  53. template <typename ForwardIterator,
  54. typename ItemToRankMap,
  55. typename SizeType>
  56. void bucket_sort(ForwardIterator begin,
  57. ForwardIterator end,
  58. ItemToRankMap rank,
  59. SizeType range = 0)
  60. {
  61. #ifdef BOOST_GRAPH_PREFER_STD_LIB
  62. std::stable_sort(begin, end, rank_comparison<ItemToRankMap>(rank));
  63. #else
  64. typedef std::vector
  65. < typename boost::property_traits<ItemToRankMap>::key_type >
  66. vector_of_values_t;
  67. typedef std::vector< vector_of_values_t > vector_of_vectors_t;
  68. if (!range)
  69. {
  70. rank_comparison<ItemToRankMap> cmp(rank);
  71. ForwardIterator max_by_rank = std::max_element(begin, end, cmp);
  72. if (max_by_rank == end)
  73. return;
  74. range = get(rank, *max_by_rank) + 1;
  75. }
  76. vector_of_vectors_t temp_values(range);
  77. for(ForwardIterator itr = begin; itr != end; ++itr)
  78. {
  79. temp_values[get(rank, *itr)].push_back(*itr);
  80. }
  81. ForwardIterator orig_seq_itr = begin;
  82. typename vector_of_vectors_t::iterator itr_end = temp_values.end();
  83. for(typename vector_of_vectors_t::iterator itr = temp_values.begin();
  84. itr != itr_end; ++itr
  85. )
  86. {
  87. typename vector_of_values_t::iterator jtr_end = itr->end();
  88. for(typename vector_of_values_t::iterator jtr = itr->begin();
  89. jtr != jtr_end; ++jtr
  90. )
  91. {
  92. *orig_seq_itr = *jtr;
  93. ++orig_seq_itr;
  94. }
  95. }
  96. #endif
  97. }
  98. template <typename ForwardIterator, typename ItemToRankMap>
  99. void bucket_sort(ForwardIterator begin,
  100. ForwardIterator end,
  101. ItemToRankMap rank)
  102. {
  103. bucket_sort(begin, end, rank, 0);
  104. }
  105. template <typename ForwardIterator>
  106. void bucket_sort(ForwardIterator begin,
  107. ForwardIterator end
  108. )
  109. {
  110. bucket_sort(begin, end, identity_property_map());
  111. }
  112. } //namespace boost
  113. #endif //__BUCKET_SORT_HPP__