题目描述:
对于一个int数组,请编写一个归并排序算法,对数组元素排序。
给定一个int数组A及数组的大小n,请返回排序后的数组。
测试用例:
输入:[1,2,3,5,2,3],6
输出:[1,2,2,3,3,5]
解题思路:
1、递归分解数组
【1,2,3,5,2,3】
【1,2,3】 【5,2,3】
【1,2】【3】 【5,2】【3】
【1】【2】【3】【5】【2】【3】
2、合并成有序的数组
【1】【2】【3】【5】【2】【3】
【1,2】【3】【5,2】【3】
【1,2,3】【2,3,5】
【1,2,2,3,3,5】
代码:
class MergeSort {
public:
int* mergeSort(int* A, int n) {
// O(n*logn)
// 如何分隔
// 如何合并2个有序的序列
mergesort(A,0,n-1);
return A;
}
void mergesort(int* A,int left,int right){
if(left < right){
int mid = (right + left)/2;
mergesort(A,left,mid);
mergesort(A,mid+1,right);
merge(A,left,mid,right);
}
}
void merge(int* A,int left,int mid,int right){
int T[right-left+1];
int temp=0;
int i=left,j=mid+1;
while((i < mid+1) && (j <= right)){
if(A[i] <= A[j]) T[temp++] = A[i++];
else T[temp++] = A[j++];
}
while(i<mid+1){
T[temp++] = A[i++];
}
while(j<=right){
T[temp++] = A[j++];
}
for(int i=0;i<temp;i++) A[left+i] = T[i];
}
};