归并排序c++设置升序降序
时间: 2023-07-17 09:09:59 浏览: 100
归并排序是一种稳定的排序算法,它可以用于升序或降序排序。下面是一个示例代码,演示了如何使用归并排序算法来实现升序和降序排序:
```cpp
#include <iostream>
#include <vector>
// 合并两个有序数组
void merge(std::vector<int>& arr, int left, int mid, int right, bool ascending) {
int n1 = mid - left + 1;
int n2 = right - mid;
std::vector<int> L(n1), R(n2);
for (int i = 0; i < n1; i++) {
L[i] = arr[left + i];
}
for (int j = 0; j < n2; j++) {
R[j] = arr[mid + 1 + j];
}
int i = 0, j = 0, k = left;
// 升序排序
if (ascending) {
while (i < n1 && j < n2) {
if (L[i] <= R[j]) {
arr[k] = L[i];
i++;
} else {
arr[k] = R[j];
j++;
}
k++;
}
}
// 降序排序
else {
while (i < n1 && j < n2) {
if (L[i] >= R[j]) {
arr[k] = L[i];
i++;
} else {
arr[k] = R[j];
j++;
}
k++;
}
}
while (i < n1) {
arr[k] = L[i];
i++;
k++;
}
while (j < n2) {
arr[k] = R[j];
j++;
k++;
}
}
// 归并排序函数
void mergeSort(std::vector<int>& arr, int left, int right, bool ascending) {
if (left < right) {
int mid = left + (right - left) / 2;
mergeSort(arr, left, mid, ascending);
mergeSort(arr, mid + 1, right, ascending);
merge(arr, left, mid, right, ascending);
}
}
// 打印数组函数
void printArray(const std::vector<int>& arr) {
for (int i = 0; i < arr.size(); i++) {
std::cout << arr[i] << " ";
}
std::cout << std::endl;
}
int main() {
std::vector<int> arr = {5, 2, 8, 4, 1};
std::cout << "原始数组: ";
printArray(arr);
// 使用归并排序进行升序排序
mergeSort(arr, 0, arr.size() - 1, true);
std::cout << "升序排序结果: ";
printArray(arr);
// 使用归并排序进行降序排序
mergeSort(arr, 0, arr.size() - 1, false);
std::cout << "降序排序结果: ";
printArray(arr);
return 0;
}
```
在上面的代码中,`merge` 函数用于合并两个有序数组。`mergeSort` 函数使用归并排序算法对数组进行排序。通过传递 `true` 或 `false` 给 `ascending` 参数,可以设置排序的升序或降序。`printArray` 函数用于打印数组的元素。
注意:这只是一个简单的示例代码,为了方便理解,数组元素直接在代码中定义。在实际应用中,你可能需要修改代码以适应不同的输入方式。
阅读全文