Bottom-up mergesort

//自底向上的归并排序

import edu.princeton.cs.algs4.*;

public class MergeBU
{
    private static Comparable[] aux;
    
    public static void merge(Comparable[] a, int lo, int mid, int hi)
    {
        int i = lo, j = mid + 1;
        
        for(int k = lo; k <= hi; k++)//复制
            aux[k] = a[k];
    
        for(int k = lo; k <= hi; k++)//把a[]填满就退出循环
        {
            if(i > mid)//左侧已用完,则只能从右侧拿
                a[k] = aux[j++];
            else if(j > hi)//右侧已用完,则只能从左侧拿
                a[k] = aux[i++];
            else if(less(aux[i], aux[j]))//当前左侧比右侧小,从右侧拿
                a[k] = aux[i++];
            else//反之当前右侧比左侧小,从左侧拿
                a[k] = aux[j++];
        }
    }
    
    public static void sort(Comparable[] a)
    {
        int N = a.length;
        aux = new Comparable[N];
        for(int sz = 1; sz < N; sz+=sz)//每次merge的大小size,从1开始到小于N的最大size
            for(int lo = 0; lo < N-sz; lo += sz+sz)//左sz和右sz进行归并,所以lo+=2sz
            merge(a, lo, lo+sz-1, Math.min(lo+sz+sz-1, N-1));//取min是怕最后边界的那个lo+sz+sz-1超过数组大小N-1,防止数组越界
    }
    
    private static boolean less(Comparable v, Comparable w)
    {
        return v.compareTo(w) < 0;
    }
    
    private static void show(Comparable[] a)
    {
        for(int i = 0; i < a.length; i++)
            StdOut.print(a[i] + " ");
        StdOut.println();
    }
    
    public static boolean isSorted(Comparable[] a)
    {
        for(int i = 1; i < a.length; i++)
            if(less(a[i], a[i-1])) return false;
        return true;
    }
    
    public static void main(String[] args)
    {
        String[] a = In.readStrings();//CTRL + d
        sort(a);
        assert isSorted(a);
        show(a);
    }
}

猜你喜欢

转载自www.cnblogs.com/w-j-c/p/9109316.html