[leetcode]347. Top K Frequent Elements -- JavaScript 代码

原创 2016年08月29日 16:09:40
/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number[]}
 */
var topKFrequent = function(nums, k) {
    function Node(n,c){
        this.num = n;
        this.count = c;
    }
    var len = nums.length;
    var dataMap = {};
    for(var i=0;i<len;i++){
        if(dataMap[nums[i]] === undefined){
            dataMap[nums[i]] = new Node(nums[i],1);
        }else{
            dataMap[nums[i]].count++;
        }
    }
    var array = [];
    for(var key in dataMap){
        array.push(dataMap[key]);
    }
    array.heapSort();
    len = array.length;
    var ret=[];
    for(i=0;i<k;i++){
        var j = len-1-i;
        ret.push(array[j].num);
    }
    return ret;
};
// 堆排序算法:大根堆调整出的其实是从小到大排列的数组。

Array.prototype.buildMaxHeap=function(){
    for(var i=Math.floor(this.length/2)-1;i>=0;i--){
        this.heapAdjust(i,this.length);
    }
};

Array.prototype.swap=function(i,j){
    var tmp=this[i];
    this[i]=this[j];
    this[j]=tmp;
};

Array.prototype.heapSort=function(){
    this.buildMaxHeap();
    for(var i=this.length-1;i>0;i--){
        this.swap(0,i);
        this.heapAdjust(0,i);
    }
    return this;
};

Array.prototype.heapAdjust=function(i,j){
    var largest=i;
    var left=2*i+1;
    var right=2*i+2;
    if(left<j&&this[largest].count<this[left].count){
        largest=left;
    }
    if(right<j&&this[largest].count<this[right].count){
        largest=right;
    }
    if(largest!=i){
        this.swap(i,largest);
        this.heapAdjust(largest,j);
    }
};

这道题代码比较长,需要注意的部分包括3个:
第一,使用Node类型来保存数据,不仅需要保存元素,还需要保存元素出现的次数。
第二,使用map来记录和整理原始数据nums。
第三,使用堆排序来保证时间复杂度符合要求。

版权声明:原创文章~盗版必究~~ 举报

相关文章推荐

[leetcode]347. Top K Frequent Elements -- JavaScript 代码

[leetcode]347. Top K Frequent Elements -- JavaScript 代码

leetcode 347. Top K Frequent Elements

1.题目Given a non-empty array of integers, return the k most frequent elements. 一个非空数组,要求返回前K个最频繁出现的元...

精选:深入理解 Docker 内部原理及网络配置

网络绝对是任何系统的核心,对于容器而言也是如此。Docker 作为目前最火的轻量级容器技术,有很多令人称道的功能,如 Docker 的镜像管理。然而,Docker的网络一直以来都比较薄弱,所以我们有必要深入了解Docker的网络知识,以满足更高的网络需求。

Leetcode 347.Top K Frequent Elements

Given a non-empty array of integers, return the k most frequent elements. 题目链接:Top K Frequent El...

LeetCode 347. Top K Frequent Elements

/***************************************************************************** * * Given a non-...

LeetCode No.347 Top K Frequent Elements

LeetCode No.347 Top K Frequent Elements

【leetcode】347. Top K Frequent Elements

Difficulty:medium Given a non-empty array of integers, return the k most frequent elements. Fo...

【leetcode】347. Top K Frequent Elements

一、题目描述 Given a non-empty array of integers, return the k most frequent elements. For example, Giv...

LeetCode:347. Top K Frequent Elements

LeetCode:347. Top K Frequent ElementsGiven a non-empty array of integers, return the k most frequent...

[leetcode] 347. Top K Frequent Elements

Given a non-empty array of integers, return the k most frequent elements. For example, Given [1,1,1...

Leetcode 347. Top K Frequent Elements

Given a non-empty array of integers, return the k most frequent elements.For example, Given [1,1,1,...
返回顶部
收藏助手
不良信息举报
您举报文章:深度学习:神经网络中的前向传播和反向传播算法推导
举报原因:
原因补充:

(最多只允许输入30个字)