C++ sorting and keeping track of indexes

by

Using C++ 11 lambdas:

#include <iostream>
#include <vector>
#include <numeric>      // std::iota
#include <algorithm>    // std::sort, std::stable_sort

using namespace std;

template <typename T>
vector<size_t> sort_indexes(const vector<T> &v) {

  // initialize original index locations
  vector<size_t> idx(v.size());
  iota(idx.begin(), idx.end(), 0);

  // sort indexes based on comparing values in v
  // using std::stable_sort instead of std::sort
  // to avoid unnecessary index re-orderings
  // when v contains elements of equal values 
  stable_sort(idx.begin(), idx.end(),
       [&v](size_t i1, size_t i2) {return v[i1] < v[i2];});

  return idx;
}

Now you can use the returned index vector in iterations such as

for (auto i: sort_indexes(v)) {
  cout << v[i] << endl;
}

You can also choose to supply your original index vector, sort function, comparator, or automatically reorder v in the sort_indexes function using an extra vector.

More Related Contents:

  1. Sorting a vector of custom objects
  2. What’s the most efficient way to erase duplicates and sort a vector?
  3. How do I sort a std::vector by the values of a different std::vector?
  4. problem sorting using member function as comparator
  5. Sort list using STL sort function
  6. Sorting a vector in descending order
  7. how do you insert the value in a sorted vector?
  8. c++ sort keeping track of indices [duplicate]
  9. How to use std::sort with a vector of structures and compare function?
  10. Sorting a std::vector by the string?
  11. Performance of qsort vs std::sort?
  12. How can I sort an STL map by value?
  13. Is std::pair ordering well-defined?
  14. Is there support in C++/STL for sorting objects by attribute?
  15. How to sort an STL vector?
  16. Which type of sorting is used in the std::sort()?
  17. What is wrong with the following code which calculates a sorted array [closed]
  18. Why is template argument deduction disabled with std::forward?
  19. Is the order of iterating through std::map known (and guaranteed by the standard)?
  20. How to get a random element from a C++ container?
  21. Multidimensional std::array [duplicate]
  22. How to choose between map and unordered_map?
  23. Advantages of std::for_each over for loop
  24. C++ equivalent of StringBuffer/StringBuilder?
  25. Does std::stack expose iterators?
  26. adding list items or nodes in linked list
  27. Why does a push_back on an std::list change a reverse iterator initialized with rbegin?
  28. Using std::map where V has no usable default constructor
  29. Comparing 3 modern c++ ways to convert integral values to strings
  30. “Proper” way to store binary data with C++/STL

Leave a Comment