归并算法是一个非常经典的分治算法,和快速排序有些类似,都是将问题分解成规模更小的子问题,分别解决。但是快速排序的子问题求解完成之后就是最优解,无需进行处理。归并算法需要对分别排序完成的子序列进行合并操作。
合并操作非常简单,定义如下:每次取数组a和b第一个元素中较小者放入新的队列,直到有个队列为空,然后将另一个队列中的元素全部放入新的队列。
当问题分解成只剩下一个元素的子问题时,必然有序,成为问题分解的边界。
代码如下:
package dataStructureAndAlgorithms;
public class MergeSort {
//temp 传进来是省去每次重新new的开销
public static void merge(int[] a, int first, int mid, int last, int[] temp) {
int index1 = first;
int index2 = mid + 1;
int k = first;
while(index1 <= mid && index2 <= last){
if(a[index1]<a[index2]){
temp[k++] = a[index1++];
}else{
temp[k++] = a[index2++];
}
}
//第一个队列剩余
while(index1<=mid){
temp[k++] = a[index1++];
}
//第二个队列剩余
while(index2<=last){
temp[k++] = a[index2++];
}
for(int i=first;i<=last;i++){
a[i] = temp[i];
}
}
public static void mergeSort(int[] a, int first, int last,int[] temp){
if(first<last){
int mid = (first + last)/2;
mergeSort(a, first, mid,temp);
mergeSort(a, mid+1, last,temp);
merge(a, first, mid, last, temp);
}
}
public static void display(int[] array){
for(int i=0;i<array.length;i++){
System.out.print(array[i] + " ");
}
System.out.println();
}
public static void main(String[] args) {
int[] originArray = {4,2,8,9,5,7,6,1,3};
int[] array = new int[originArray.length];
int[] temp = new int[array.length];
System.arraycopy(originArray, 0, array, 0, originArray.length);
System.out.println("\n\n未排序数组顺序为:");
display(array);
System.out.println("-----------------------");
mergeSort(array, 0, array.length-1, temp);
System.out.println("-----------------------");
System.out.println("经过归并排序后的数组顺序为:");
display(array);
}
}
输出:
未排序数组顺序为:
4 2 8 9 5 7 6 1 3
-----------------------
-----------------------
经过归并排序后的数组顺序为:
1 2 3 4 5 6 7 8 9