347. Top K Frequent Elements

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

For example,
Given [1,1,1,2,2,3] and k = 2, return [1,2].

Note:
You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

这道题要找到出现次数从多到少排的前k个元素。
首先我想到的是使用一个map来保存所有的元素出现的次数,然后遍历这个map找到前k个。
怎么遍历这个map呢,首先想到的是把它读到一个数组里然后排序:

var topKFrequent = function(nums, k) {
    var map = {};
    var result = [];
    for (var i = 0;i < nums.length;i++) {
        if (map[nums[i]]===undefined)
            map[nums[i]]=1;
        else {
            map[nums[i]]++;
        }
    }
    for (var item in map) {
        result.push([item,map[item]]);
    }
    result.sort(function(a,b){
        return -(a[1]-b[1]);
    });
    for (i = 0; i < k;i++) {
        result[i] = parseInt(result[i][0]);
    }
    return result.slice(0,k);
};

不过还有个更好的办法,所有元素出现的顺序都不会大于n,所以我们建一个长度为n+1的数组,用下标来表示没个元素出现的次数,由于会有多个元素出现相同的次数,所以这会是个二维数组。
最后从后往前遍历这个数组,取出k个数就好。
这样就不用排序了。

var topKFrequent = function(nums, k) {
    var map = {};
    var result = [];
    var res = [];
    for (var i = 0;i < nums.length;i++) {
        if (map[nums[i]]===undefined)
            map[nums[i]]=1;
        else {
            map[nums[i]]++;
        }
    }
    for (var item in map) {
        if (result[map[item]]!==undefined) 
            result[map[item]].push(item);
        else
            result[map[item]] = [item];
    }
    var num = result.length-1;
    var count = 0;
    while(count!==k) {
        if (result[num]!==undefined) {
            for (i=0; i

你可能感兴趣的:(347. Top K Frequent Elements)