Counting swaps using std::sort

user_5594299 0
Viewed 231

Is there a portable, minimal-overhead way to count the number of swap operations performed during a std::sort in C++? I would like to do this because I need to compute the sign of the permutation used to sort the list, and I was wondering if there was a way to reuse std::sort for this rather than writing my own sorting function.

sorting c++
1 Answers

I tried to answer real quick by making a wrapper/ custom type to overload std::swap...and then ran into the fact that for super small vectors swap isn't called...following the link in the comments so attempt 2 added a counter for the move_constructor.

I can't say this is a minimal overhead solution and you're probably better off writing your own sorting function if you need the exact number of swap operations.

#include <iostream>
#include <vector>
#include <algorithm>
#include <random>

struct A{
  static int swap_count;
  static int move_constructor_count;
  int a;
  A(int _a): a(_a) {}
  bool operator<(const A& other) const{
    return this->a < other.a;
  }
  A(const A&other): a(other.a) {move_constructor_count++;}
};
int A::swap_count = 0;
int A::move_constructor_count = 0;

namespace std{
    template<>
    void swap(A& lhs, A& rhs) {
       A::swap_count++;
       std::swap(lhs.a, rhs.a);
    }
}


int main() {
  std::default_random_engine gen;
  std::uniform_int_distribution<int> dis(1,100);

  std::vector<A> test;
  for(int _=0;_<10;_++) test.emplace_back(dis(gen)); //fill a vector randomly

  A::move_constructor_count = 0; // emplace calls move constructor
  std::sort(test.begin(), test.end());
  std::cout << "after sort1: swap count:" << A::swap_count << " move count: " << A::move_constructor_count << std::endl;


  // arbitrary way to fill a large test vector
  std::vector<A> test2;
  for(int _=0;_<1000;_++) test2.emplace_back(dis(gen)); //fill a vector randomly
    
  A::move_constructor_count = 0;
  A::swap_count = 0;
  std::sort(test2.begin(), test2.end());
  std::cout << "after sort2: swap count:" << A::swap_count << " move count: " << A::move_constructor_count << std::endl;

}

gave me

after sort1: swap count:0 move count: 9
after sort2: swap count:1806 move count: 999
user_92374860
Related