归并排序法
- 将数组分成两半,然后分别对其排序
- 在将两部分在分别分成两半,然后对其排序
- 直到每部分只剩下一个元素
- 最后再使用递归merge两边的数组。
Merge时,需要借助三个指针。
一个指向左半部分;一个指向右半部分;一个指向辅助数组。
class Solution {
public:
// 归并排序
void merge(vector<int>& arr, int low, int mid, int high){
// vector<int> temp;
int temp[high-low+1];
// 辅助数组,临时储存要操作部分的数组
for(int i = 0; i<=high-low; i++){
// temp.push_back(arr[i+low]);
temp[i] = arr[i+low];
}
int j = low; // 左半部分指针
int k = mid+1; // 右半部分指针
for(int i = low; i <= high; i++){
if(j>mid){
arr[i] = temp[k-low];
k++;
}else if(k > high){
arr[i] = temp[j-low];
j++;
}else if(temp[j-low] < temp[k-low]){
arr[i] = temp[j-low];
j++;
}else{
arr[i] = temp[k-low];
k++;
}
}
}
void mergeSort(vector<int>& arr, int low, int high){
// 出归条件,就剩一个元素
// 通用改进方法,在递归到数量小到一定程度时,使用插入排序
// 因为插入排序(n^2和nlogn差不多),且有序的概率更高
if(low >= high){
return;
}
// 切分为[low,mid], [mid+1,high]
int mid = (low+high)/2;
mergeSort(arr, low, mid);
mergeSort(arr, mid+1, high);
// 改进方法,如果arr[mid]<=arr[mid+1],说明已经有序了(特别是对于有序数组)
if(arr[mid]>arr[mid+1]){
merge(arr,low,mid,high);
}
}
vector<int> sortArray(vector<int>& nums) {
// [low,high]
int low = 0;
int high = nums.size() - 1;
mergeSort(nums, low, high);
return nums;
}
};
改进方法:
- 通用改进方法,在递归到数量小到一定程度时,使用插入排序。因为插入排序(n^2和nlogn差不多),且有序的概率更高。
- 对于merge, 改进方法,如果arr[mid]<=arr[mid+1],说明已经有序了(特别是对于有序数组),这时不需要merge