一个优化的快速排序

快速排序是一个最差时间复杂度为O(n²)的排序算法,这种情况通常出现在选择的轴值(pivot)不能将数组划分为两个长度相等的子数组的时候,比如数组逆序排列的时候,如果选择第一个数作为轴值,划分的子数组的大小分别为0和n-1,此时算法的性能最差。

一个较好的办法是“三数取中”,查看当前数组的第一个、中间一个和最后一个位置的数组,取其中位数,以此来降低轴值选择得不好的可能性。

JavaScript实现代码如下:

//作为对比,这里先给出未使用三数取中的情况
Array.prototype.swap = function (i, j) {
  var t = this[i];
  this[i] = this[j];
  this[j] = t;
};
Array.prototype.quickSortHelper = function(start,end){
    if(start>=end){
        return;
    }
    var pivotIdx = start;
    var pivot = this[pivotIdx];
    var i = start+1;
    var n = end;
    while(i<=n){
        if(this[i]<pivot){
           //比轴值小则交换
            this.swap(pivotIdx,i);
            i++;
            pivotIdx = i;
        }else{
            this.swap(n,i);
            n--;
        }
    }
    //递归地对子数组进行排序
    this.quickSortHelper(start,pivotIdx-1);
    this.quickSortHelper(pivotIdx+1,end);
}
Array.prototype.quickSort = function () {
  this.quickSortHelper(0, this.length-1);
};

测试一下算法的性能:

function test () {
    var arr = [];
    for (var i = 0; i < 20000; i++) {
        arr.push(Math.round(Math.random(i) * 100));
    }
    //为了测试,这里故意将数组按照逆序排序
    arr.sort(function(x,y){
        return y-x;
    });
    doTest(arr, 1);
}
function doTest(arr, n) {
    var tStart = (new Date()).getTime();
    var re = arr.quickSort(0,arr.length-1);
    var tEnd = (new Date()).getTime();
    console.log('快速排序使用时间是:' + (tEnd - tStart) + 'ms');
    return re;
}
test();//输出:快速排序使用时间是:168ms

排序时间为168ms。

进行三数取中后,再来看看性能如何:

//将开头、中间、结尾位置的中间一个元素交换到开头
Array.prototype.partionMedianOfThree = function(start,end){
    var mid = Math.floor(start+(end-start)/2);
    if(this[start]>this[end]){
        this.swap(start,end);
    }
    if(this[mid]>this[end]){
        this.swap(mid,end);
    }
    if(this[mid]>this[start]){
        this.swap(mid,start);
    }
}
//quickHelper中先进行三数取中,其余地方不变
Array.prototype.quickSortHelper = function(start,end){
    if(start>=end){
        return;
    }
    this.partionMedianOfThree(start,end);
    var pivotIdx = start;
    var pivot = this[pivotIdx];
    var i = start+1;
    var n = end;
    while(i<=n){
        if(this[i]<pivot){
            this.swap(pivotIdx,i);
            i++;
            pivotIdx = i;
        }else{
            this.swap(n,i);
            n--;
        }
    }
    this.quickSortHelper(start,pivotIdx-1);
    this.quickSortHelper(pivotIdx+1,end);
}

结果现在的排序时间缩短到:22ms。在这种情况下的性能提升是非常显著的。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Ba la la la ~ 读者朋友们,你们好啊,又到了冷锋时间,话不多说,发车! 1.冒泡排序(Bub...
    王饱饱阅读 1,816评论 0 7
  • quicksort可以说是应用最广泛的排序算法之一,它的基本思想是分治法,选择一个pivot(中轴点),将小于pi...
    黎景阳阅读 471评论 0 1
  • 天色昏暗。 梦想这词,太奢侈,妈蛋。 快乐与高薪,你总得有一样吧?如果可以两者兼得自然是美事,但生活中多的是不可兼...
    大尾巴猫咪阅读 365评论 0 0
  • 章节次第,经兹证正,文义条贯,章法显明,细心求之,自达原旨。 中庸之德必本于孝悌。 我父母管我很严,也算是保护很好...
    杨寅成阅读 159评论 0 0
  • 理解对象 1、定义属性(两种) 2、属性探测(两种) 用 in 操作符 和 hasOwnProperty 3、删除...
    AnnQi阅读 167评论 0 0