跟JDK学算法之快速排序

看源码时专门瞄了下排序算法,跟着学一下。JDK的排序是 插入排序、快排揉一起的,下面学下快速排序。在JDK中当待排序的元素个数小于 ** 286 ** 时使用快速排序,当待排序的元素个数小于 ** 47 ** 时使用插入排序。JDK源码类:DualPivotQuicksort.java
本文地址://www.greatytc.com/p/a7ceb2dc62ca

由于JDK的算法写的怪怪的(个人感觉),所以不参照它的代码,只参照它的思想。

对于快速排序,JDK先采用双轴快排,即:取5个元素,按顺序排好,然后取第二个和第四个元素作为轴开始排序。如果这五个元素恰好相等,则使用中间的元素作为轴开始快排(传统的快排)

  • 先来看下传统的快排:单轴快排,轴的选择有很多种,第一个元素,最后一个元素,中心位置元素,随机位置元素,取首尾中3个元素中的中间值等,也可以参照JDK取5个元素的中值作为轴。此处以取首尾中3个元素中的中间值为例
public static void basicSort(int[] array) {
    if (array == null || array.length < 2) {
        return;
    }
    basicSort(array, 0, array.length - 1);
}

private static void basicSort(int[] array, int low, int high) {
    if (low >= high) {
        return;
    }
    // 这里通常判断下待排序元素的个数,比如小于47个使用插入排序,以提高性能
    // .....

    // 单轴快排,核心在取轴这
    int position = basicPosition(array, low, high);
    basicSort(array, low, position - 1);
    basicSort(array, position + 1, high);
}

private static int basicPosition(int[] array, int low, int high) {
    // 初始化轴数据,获取轴数据并放到数组第一个位置
    initPivotData(array, low, high);
    
    int pivotData = array[low];     // 轴数据
    int begin = low, end = high;
    while (begin < end) {
        while (array[begin] <= pivotData && begin < high) {
            begin++;
        }
        while (array[end] > pivotData && end >= low) {
            end--;
        }
        if (begin < end) {
            swap(array, begin, end);
        }
        printArray(array);
    }
    if (end != low) {
        swap(array, low, end);
        printArray(array);
    }
    return end;
}
// 这里将3各元素做升序排序,并将中值交换到数组第一个元素处
private static void initPivotData(int[] array, int low, int high) {
    int mid = low + (high - low) / 2;
    
    if (array[mid] > array[high]) {
        swap(array, mid, high);
    }
    if (array[low] > array[high]) {
        swap(array, low, high);
    }
    if (array[mid] > array[low]) {
        swap(array, mid, low);
    }
}

private static void swap(int[] array, int i, int j) {
    int temp = array[i];
    array[i] = array[j];
    array[j] = temp;
}
  • 下面看下JDK的双轴快排,选两个轴,排序方式,比轴1小的数据放到轴1左侧,比轴2大的数据放到轴2右侧,其余的数据在轴1和轴2之间
public static void dualPivotSort(int[] array) {
    dualPivotSort(array, 0, array.length-1);
}

// 参照JDK取轴的方法,该方法未使用
private static void initdualPivotData(int[] array, int low, int high) {
    int length = high - low + 1;    // 待排序元素的个数
    int seventh = (length >> 3) + (length >> 6) + 1;    // 7 等份
    
    // 取5个节点
    int e3 = (low + high) >>> 1; // The midpoint
    int e2 = e3 - seventh;
    int e1 = e2 - seventh;
    int e4 = e3 + seventh;
    int e5 = e4 + seventh;

    // 对这5各元素做插入排序,然后去e2和e4作为轴
}

private static void dualPivotSort(int[] array, int lowIndex, int highIndex) {

    if (highIndex <= lowIndex) return;
    
//        initdualPivotData(array, lowIndex, highIndex);
    
    int pivot1 = array[lowIndex];
    int pivot2 = array[highIndex];

    // 确保 pivot1 < pivot2
    if (pivot1 > pivot2) {
        swap(array, lowIndex, highIndex);
        pivot1 = array[lowIndex];
        pivot2 = array[highIndex];
    }
    else if (pivot1 == pivot2) {
        while (pivot1 == pivot2 && lowIndex < highIndex){
            lowIndex++;
            pivot1 = array[lowIndex];
        }
    }

    int i = lowIndex + 1;
    int lt = lowIndex + 1;
    int gt = highIndex - 1;

    while (i <= gt) {
        if (array[i] < pivot1) {
            // 小于轴1,则将数据交换到轴的左侧
            swap(array, i++, lt++);
        }
        else if (pivot2 < array[i]) {
            // 大于轴2,则将数据交换到轴的右侧
            swap(array, i, gt--);
        }
        else {
            i++; // 其他不处理,索引向后移动1
        }
    }

    /* 都交换之后
     * 将轴1交换到中间部分的左侧;轴2交换到中间部分的右侧
     * 这样确保左侧都小于轴1,右侧都大于轴2
     */
    swap(array, lowIndex, --lt);
    swap(array, highIndex, ++gt);

    // 普通的快排,只需对轴两侧递归处理
    // 这里是双轴快排,所以要对左侧、中间、右侧分别递归处理
    dualPivotSort(array, lowIndex, lt - 1);
    dualPivotSort (array, lt + 1, gt - 1);
    dualPivotSort(array, gt + 1, highIndex);
}
  • 快速排序还有一种方式用的比较多,3路排序。该方案也是选一个轴,比轴小的数据在轴的左侧,比轴大的数据在轴的右侧,中间是与轴相等的数据。
public static void threeWaySort(int[] array) {
    threeWaySort(array, 0, array.length-1);
}

private static void threeWaySort(int[] array, int low, int high) {
    
    if (high <= low)
        return;

    // 初始化轴数据,获取轴数据并放到数组第一个位置(使用第一种快排的方式获取)
    initPivotData(array, low, high);
    
    int lt = low;   // 记录最后一次发现低值的位置
    int gt = high;  // 记录最后一次发现高值的位置
    int i = low + 1;// 用于从左侧遍历到右侧

    int pivotIndex = low;
    int pivotValue = array[pivotIndex];

    while (i <= gt) {
        // 小于轴,交换到左侧
        if (array[i] < pivotValue) {
            swap(array, i++, lt++);
        }
        // 大于轴,交换到右侧
        else if (pivotValue < array[i]) {
            swap(array, i, gt--);
        }
        else {
            i++;    // 相等不做处理
        }
    }

    // 这里与单轴快排一样
    threeWaySort(array, low, lt - 1);
    threeWaySort(array, gt + 1, high);
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,126评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,254评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,445评论 0 341
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,185评论 1 278
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,178评论 5 371
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,970评论 1 284
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,276评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,927评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,400评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,883评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,997评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,646评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,213评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,204评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,423评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,423评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,722评论 2 345

推荐阅读更多精彩内容