工具函数之取数组的topN和bottomN

最新推荐文章于 2025-08-15 16:10:12 发布

原创最新推荐文章于 2025-08-15 16:10:12 发布 · 993 阅读

0 ·

CC 4.0 BY-SA版权

javascript 专栏收录该内容

121 篇文章

订阅专栏

本文介绍了一种改进的Shell排序算法实现，并基于此实现了获取数组中Top-N和Bottom-N元素的功能。通过调整排序步长并采用插入排序进行局部优化，Shell排序能够更高效地对数组进行排序。

    var C = {};
    C.shellSort = function(arr) {
        var step = 5;
        while (step > 0) {
            for (var i = step;i<arr.length;i++) {
                var j = i-step;
                var key = arr[i];
                while (j>=0 && key<arr[j]) {
                    arr[j+step] = arr[j];
                    j-=step;
                }
                arr[j+step] = key;
            }
            step = Math.floor(step/2);
        }
        return arr;
    }

    C.topN = function(arr,n) {
        if(!arr) {
            return false;
        }
        if (!n) {
            return
        }
        return C.creatorN(arr,n,'top');
    }

    C.bottomN = function(arr,n) {
        return C.creatorN(arr,n,'bottom');
    }

    C.creatorN = function(arr,n,type) {
        var result = C.shellSort(arr);
        return type==='top' ? result.slice(0,n) : result.slice(arr.length-n);
    }