自顶向下
1 #include <iostream> 2 #include <algorithm> 3 #include "InsertionSort.h" 4 5 using namespace std; 6 7 template<typename T> 8 // 将arr[l...mid]和arr[mid+1...r]两部分进行归并 9 void __merge(T arr[] , int l, int mid, int r){ 10 T aux[r-l+1]; 11 // 将数组复制一份到aux[] 12 for( int i = l ; i <= r ; i ++ ) 13 aux[i-l] = arr[i]; 14 // 初始化,i、j指向左、右半部分的起始索引位置 15 int i = l, j = mid + 1; 16 for( int k = l ; k <= r ; k ++ ){ 17 // 如果左半部分已处理完毕 18 if( i > mid){ 19 arr[k] = aux[j-l]; 20 j++; 21 } 22 // 如果右半部分已处理完毕 23 else if( j > r){ 24 arr[k] = aux[i-l]; 25 i ++; 26 } 27 // 左半部分所指元素 < 右半部分所指元素 28 else if( aux[i-l] < aux[j-l] ){ 29 arr[k] = aux[i-l]; 30 i ++; 31 } 32 // 左半部分所指元素 >= 右半部分所指元素 33 else{ 34 arr[k] = aux[j-l]; 35 j++; 36 } 37 } 38 } 39 40 // 递归使用归并排序,对arr[l...r]的范围进行排序 41 template<typename T> 42 void __mergeSort(T arr[] , int l, int r){ 43 44 if( l >= r) 45 return; 46 47 int mid = (l+r)/2; 48 __mergeSort(arr,l,mid); 49 __mergeSort(arr,mid+1,r); 50 // 优化,前后两部分有序时不归并 51 if( arr[mid] > arr[mid+1] ) 52 __merge(arr, l, mid, r); 53 } 54 55 template<typename T> 56 void mergeSort(T arr[] , int n){ 57 __mergeSort( arr , 0 , n-1 ); 58 }
自底向上
1 //自底向上归并,可对链表排序 2 template<typename T> 3 void mergeSortBU(T arr[], int n){ 4 for( int sz = 1 ; sz <= n ; sz += sz ) 5 for( int i = 0 ; i + sz < n ; i += sz + sz ) 6 // 对 arr[i...i+sz-1] 和 arr[i+sz...i+2*sz-1] 进行归并 7 __merge( arr , i , i + sz - 1 , min(i + sz + sz -1,n-1)); 8 }
应用:求逆序对