added comp argument and test for stability

This commit is contained in:
Ahmet Inan 2024-03-08 13:00:40 +01:00
commit 389fcc3bd8
2 changed files with 37 additions and 18 deletions

View file

@ -22,25 +22,27 @@ int main()
typedef std::default_random_engine generator;
typedef std::uniform_int_distribution<int> distribution;
auto rand = std::bind(distribution(1, MAX_N), generator(seed));
int a[MAX_N], b[MAX_N], c[MAX_N], d[MAX_N], e[MAX_N], f[MAX_N], g[MAX_N];
CODE::MergeSort<int, MAX_N> merge_sort;
typedef std::pair<int, int> Pair;
Pair a[MAX_N], b[MAX_N], c[MAX_N];
int d[MAX_N], e[MAX_N];
auto comp = [](Pair a, Pair b){ return a.second < b.second; };
CODE::MergeSort<Pair, MAX_N> merge_sort;
for (int loop = 0; loop < 1000000; ++loop) {
int size = rand();
for (int i = 0; i < size; ++i)
a[i] = b[i] = c[i] = d[i] = e[i] = rand();
std::sort(a, a+size);
CODE::insertion_sort(b, size);
a[i].first = b[i].first = c[i].first = e[i] = i;
for (int i = 0; i < size; ++i)
a[i].second = b[i].second = c[i].second = d[i] = rand();
std::stable_sort(a, a+size, comp);
CODE::insertion_sort(b, size, comp);
for (int i = 0; i < size; ++i)
assert(a[i] == b[i]);
merge_sort(c, size);
merge_sort(c, size, comp);
for (int i = 0; i < size; ++i)
assert(a[i] == c[i]);
CODE::insertion_sort(e, d, size);
for (int i = 0; i < size; ++i)
f[i] = i;
std::stable_sort(f, f+size, [d](int i, int j){ return d[i] < d[j]; });
CODE::insertion_sort(g, e, size);
for (int i = 0; i < size; ++i)
assert(f[i] == g[i]);
assert(a[i].first == e[i]);
}
std::cerr << "Sorting regression test passed!" << std::endl;
return 0;