mirror of
https://github.com/aicodix/code.git
synced 2026-04-27 14:30:36 +00:00
added comp argument and test for stability
This commit is contained in:
parent
ac5fc8deea
commit
389fcc3bd8
2 changed files with 37 additions and 18 deletions
|
|
@ -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;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue