排序算法之——归并排序(两种方法及其优化)
本文将围绕代码从多个方面分析归并算法,归并的操作很简单,稍加思考便能深刻理解。
1、算法思想:
要将一个数组排序,可以(递归地)将数组分成两半分别排序,然后将两边归并起来。归并算法最吸引人的地方是它能保证将任意长度为N的数组排序的时间与NlgN成正比。
主要缺点是需要与N成正比的额外空间。
(示意图1)
2、原地归并的抽象方法
实现归并最直截了当的方法是将两个数组归并到第三个数组,实现的方法很简单,从左到右逐一比较两数组的第一位元素,将小的一个放入第三个数组(假设两数组已经有序),完成操作后第三个数组就是有序的。了解了思路,我们直接看代码。
1 public static void merge(Comparable[] a, int lo, int mid, int hi) {
2 int i = lo, j = mid + 1;
3 for (int k = lo; k <= hi; k++) {
4 aux[k] = a[k];
5 }
6 for (int k = lo; k <= hi; k++) {
7 if (i > mid) {
8 a[k] = aux[j++];
9 } else if (j > hi) {
10 a[k] = aux[i++];
11 } else if (less(aux[i], aux[j])) {
12 a[k] = aux[i++];
13 } else {
14 a[k] = aux[j++];
15 }
16 }
17
18 }
主要操作就是第二个for循环里的四个判断:
1、数组1走完(将数组2当前元素放入数组3)
2、数组2走完(将数组1当前元素放入数组3)
3、数组1当前元素小于数组2当前元素(将数组1当前元素放入数组3)
4、数组2当前元素小于等于数组1当前元素(将数组2当前元素放入数组3)
(示意图2:将数组1和组2归并到组3)
3、自顶向下的归并排序
如果能将两个子数组排序,就能通过并归两个子数组来对整个数组排序,这一切是通过递归实现的,也叫递归归并。直接看代码:
1 public class Merge{
2 private static Comparable[] aux;
3 public static void sort(Comparable[] a) {
4 aux = a.clone();// 一次性分配空间
5 sort(a,0, a.length - 1);
6 }
7
8 private static void sort(Comparable[] a,int lo, int hi) {
9 if (hi <= lo) {
10 return;
11 }
12 int mid = lo + (hi - lo) / 2;
13 sort(aux,a, lo, mid);//左半边排序
14 sort(aux,a, mid + 1, hi);//右半边排序
15 merge(a,aux,lo, mid, hi);//归并结果(参考原地归并的抽象方法)
16 }
17 }
示意图:
(示意图3)
上图只是merge方法的轨迹,sort方法也极为重要,要想理解就必须知道sort方法调用的轨迹(这里请读者自己先写出sort的轨迹再看下面的答案)
sort(a,0,7)
将左半部分排序
sort(a,0,3)
sort(a,0,1)
merge(a,0,0,1)
sort(a,2,3)
merge(a,2,2,3)
将右半部分排序
sort(a,4,7)
sort(a,4,5)
merge(a,4,4,5)
sort(a,6,7)
merge(a,6,6,7)
归并结果
merge(a,0,3,7)
4、自底向上的归并排序
我们已经知道,自顶向下采用的是递归的方法,而自底向上则是循序渐进得解决问题,采用了循环的方法。通过下图可以很容易看出两种方式的区别:
下面上代码:
1 public static void sort(Comparable[] a) {
2 int n = a.length;
3 aux = new Comparable[n];
4 for (int sz = 1; sz < n; sz = sz + sz) {
5 for (int lo = 0; lo < n - sz; lo += sz + sz) {
6 merge(a, lo, lo + sz - 1, Math.min(lo + 2 * sz - 1, n - 1));// 最后一次并归的第二个子数组可能比第一个小此时lo+2*sz-1越界
7 }
8 }
9 }
读者自行考虑自底向上方法的运行轨迹。
5、三项优化(代码在后面的代码演示中)
①对小规模子数组使用插入排序
用不同的方法处理小规模数组能改进大多递归算法的性能,在小数组上上,插入排序可能比并归排序更快。
②测试数组是否有序
根据归并排序的特点,每次归并的两个小数组都是有序的,当a[mid]<=a[mid+1]时我们可以跳过merge方法,这样并不影响排序的递归调用。
③不将元素复制到辅助数组
我们可以节省将数组复制到辅助数组的时间,这需要一些技巧。先克隆原数组到辅助数组,然后在之后的递归交换输入数组和辅助数组的角色(通过看代码更容易理解)
(画方框的为每次的输出数组)
6、代码演示(java):
1 public class Merge implements Comparable<Merge> {// 归并排序(优化前)
2 private static Comparable[] aux;
3
4 private static boolean less(Comparable v, Comparable w) {
5 return v.compareTo(w) < 0;
6 }
7
8 @Override
9 public int compareTo(Merge arg0) {
10 // TODO Auto-generated method stub
11 return 0;
12 }
13
14 public static void merge(Comparable[] a, int lo, int mid, int hi) {// 原地归并的抽象方法
15 int i = lo, j = mid + 1;
16 for (int k = lo; k <= hi; k++) {
17 aux[k] = a[k];
18 }
19 for (int k = lo; k <= hi; k++) {
20 if (i > mid) {
21 a[k] = aux[j++];
22 } else if (j > hi) {
23 a[k] = aux[i++];
24 } else if (less(aux[j], aux[i])) {
25 a[k] = aux[j++];
26 } else {
27 a[k] = aux[i++];
28 }
29 }
30 }
31
32 public static void sort(Comparable[] a) {
33 aux = new Comparable[a.length];
34 sort(a, 0, a.length - 1);
35 }
36
37 private static void sort(Comparable[] a, int lo, int hi) {
38 /*
39 * 自顶向下的并归排序 三个改进
40 */
41 if (hi <= lo) {
42 return;
43 }
44 int mid = lo + (hi - lo) / 2;
45 sort(a, lo, mid);
46 sort(a, mid + 1, hi);
47 merge(a, lo, mid, hi);
48 }
49
50 private static void exch(Comparable[] a, int j, int i) {
51 // TODO Auto-generated method stub
52 Comparable temp;
53 temp = a[j];
54 a[j] = a[i];
55 a[i] = temp;
56 }
57
58 public static void main(String[] args) {
59 Merge mg = new Merge();
60 Comparable a[] = { 8, 1, 6, 8, 4, 6, 9, 7, 1, 2, 3, 4, 8, 5, 2, 6, 4, 3, 8 };
61 mg.sort(a);
62 for (int i = 0; i < a.length; i++) {
63 System.out.print(a[i] + " ");
64 }
65 }
66 }
1 public class MergeX implements Comparable<Merge> {// 归并排序(优化后)
2 private static Comparable[] aux;
3
4 private static boolean less(Comparable v, Comparable w) {
5 return v.compareTo(w) < 0;
6 }
7
8 @Override
9 public int compareTo(Merge arg0) {
10 // TODO Auto-generated method stub
11 return 0;
12 }
13
14 public static void merge(Comparable[] a, Comparable[] aux, int lo, int mid, int hi) {// 原地归并的抽象方法
15 int i = lo, j = mid + 1;
16 // for (int k = lo; k <= hi; k++) {
17 // aux[k] = a[k];
18 // }
19 for (int k = lo; k <= hi; k++) {
20 if (i > mid) {
21 a[k] = aux[j++];
22 } else if (j > hi) {
23 a[k] = aux[i++];
24 } else if (less(aux[j], aux[i])) {
25 a[k] = aux[j++];
26 } else {
27 a[k] = aux[i++];
28 }
29 }
30 }
31
32 public static void sort(Comparable[] a) {
33 aux = a.clone();// 一次性分配空间
34 sort(a, aux, 0, a.length - 1);
35 }
36
37 private static void sort(Comparable[] a, Comparable[] aux, int lo, int hi) {
38 /*
39 * 自顶向下的并归排序 三个改进
40 */
41 // if (hi <= lo) {
42 // return;
43 // }
44 int mid = lo + (hi - lo) / 2;
45 if (hi - lo <= 7) {// 对小规模子数组使用插入排序
46 //System.out.println("insert!");
47 insertionSort(a, lo, hi);
48 return;
49 }
50 sort(aux, a, lo, mid);
51 sort(aux, a, mid + 1, hi);
52 if (!less(aux[mid + 1], aux[mid])) {// 已经有序时跳过merge(a中lo到mid mid到hi分别都是有序的)
53 System.arraycopy(aux, lo, a, lo, hi-lo+1);
54 return;
55 }
56 merge(a, aux, lo, mid, hi);
57 }
58
59 private static void insertionSort(Comparable[] a, int lo, int hi) {
60 for (int i = lo; i <= hi; i++)
61 for (int j = i; j > lo && less(a[j], a[j - 1]); j--)
62 exch(a, j, j - 1);
63 }
64
65 private static void exch(Comparable[] a, int j, int i) {
66 // TODO Auto-generated method stub
67 Comparable temp;
68 temp = a[j];
69 a[j] = a[i];
70 a[i] = temp;
71 }
72
73 public static void main(String[] args) {
74 MergeX mgx = new MergeX();
75 Comparable a[] = { 8, 1, 6, 8, 4, 6, 9,7,1, 2, 3,4,8,5,2,6,4,3,8};
76 mgx.sort(a);
77 for (int i = 0; i < a.length; i++) {
78 System.out.print(a[i] + " ");
79 }
80 }
81 }