Update wave_sort.cpp

This commit is contained in:
Swastika Gupta
2021-07-14 15:54:41 +05:30
committed by GitHub
parent b8dd0a6aae
commit c1390dcfb9

View File

@@ -4,12 +4,12 @@
* @details
* Wave Sort is a sorting algorithm that works in \f$O(nlogn)\f$ time assuming
* the sort function used works in \f$O(nlogn)\f$ time.
* @author [Swastika Gupta](https://github.com/swastyy)
* @author [Swastika Gupta](https://github.com/Swastyy)
*/
#include <algorithm> /// for std::is_sorted, std::swap
#include <cassert> /// for assert
#include <iostream> /// for io operations
#include <iostream> /// for IO operations
#include <vector> /// for std::vector
/**
@@ -29,14 +29,14 @@ namespace wave_sort {
* @returns arr the wave sorted array
*/
template <typename T>
std::vector<T> waveSort(const std::vector<T> &in_arr, int n) {
std::vector<T> waveSort(const std::vector<T> &in_arr, int64_t n) {
std::vector<T> arr(in_arr);
for (int i = 0; i < n; i++) {
for (int64_t i = 0; i < n; i++) {
arr[i] = in_arr[i];
}
std::sort(arr.begin(), arr.end());
for (int i = 0; i < n - 1; i += 2) { // swap all the adjacent elements
for (int64_t i = 0; i < n - 1; i += 2) { // swap all the adjacent elements
std::swap(arr[i], arr[i + 1]);
}
return arr;
@@ -50,34 +50,34 @@ std::vector<T> waveSort(const std::vector<T> &in_arr, int n) {
*/
static void test() {
// [10, 90, 49, 2, 1, 5, 23] return [2, 1, 10, 5, 49, 23, 90]
std::vector<int> array1 = {10, 90, 49, 2, 1, 5, 23};
std::vector<int64_t> array1 = {10, 90, 49, 2, 1, 5, 23};
std::cout << "Test 1... ";
std::vector<int> arr1 = sorting::wave_sort::waveSort(array1, 7);
const std::vector<int> o1 = {2, 1, 10, 5, 49, 23, 90};
std::vector<int64_t> arr1 = sorting::wave_sort::waveSort(array1, 7);
const std::vector<int64_t> o1 = {2, 1, 10, 5, 49, 23, 90};
assert(arr1 == o1);
std::cout << "passed" << std::endl;
// [1, 3, 4, 2, 7, 8] return [2, 1, 4, 3, 8, 7]
std::vector<int> array2 = {1, 3, 4, 2, 7, 8};
std::vector<int64_t> array2 = {1, 3, 4, 2, 7, 8};
std::cout << "Test 2... ";
std::vector<int> arr2 = sorting::wave_sort::waveSort(array2, 6);
const std::vector<int> o2 = {2, 1, 4, 3, 8, 7};
std::vector<int64_t> arr2 = sorting::wave_sort::waveSort(array2, 6);
const std::vector<int64_t> o2 = {2, 1, 4, 3, 8, 7};
assert(arr2 == o2);
std::cout << "passed" << std::endl;
// [3, 3, 3, 3] return [3, 3, 3, 3]
std::vector<int> array3 = {3, 3, 3, 3};
std::vector<int64_t> array3 = {3, 3, 3, 3};
std::cout << "Test 3... ";
std::vector<int> arr3 = sorting::wave_sort::waveSort(array3, 4);
const std::vector<int> o3 = {3, 3, 3, 3};
std::vector<int64_t> arr3 = sorting::wave_sort::waveSort(array3, 4);
const std::vector<int64_t> o3 = {3, 3, 3, 3};
assert(arr3 == o3);
std::cout << "passed" << std::endl;
// [9, 4, 6, 8, 14, 3] return [4, 3, 8, 6, 14, 9]
std::vector<int> array4 = {9, 4, 6, 8, 14, 3};
std::vector<int64_t> array4 = {9, 4, 6, 8, 14, 3};
std::cout << "Test 4... ";
std::vector<int> arr4 = sorting::wave_sort::waveSort(array4, 6);
const std::vector<int> o4 = {4, 3, 8, 6, 14, 9};
std::vector<int64_t> arr4 = sorting::wave_sort::waveSort(array4, 6);
const std::vector<int64_t> o4 = {4, 3, 8, 6, 14, 9};
assert(arr4 == o4);
std::cout << "passed" << std::endl;
}