码迷,mamicode.com
首页 > 其他好文 > 详细

前K个高频元素

时间:2018-07-18 23:14:07      阅读:110      评论:0      收藏:0      [点我收藏+]

标签:情况   出现   快排   sum   col   size   note   array   sort   

给定一个非空的整数数组,返回其中出现频率前 高的元素。

例如,

给定数组 [1,1,1,2,2,3] , 和 k = 2,返回 [1,2]

注意:

    • 你可以假设给定的 总是合理的,1 ≤ k ≤ 数组中不相同的元素的个数。
    • 你的算法的时间复杂度必须优于 O(n log n) , 是数组的大小。

 

先做一个快排(我爱qsort),然后从头开始统计元素个数,将结果记录到count和a两个数组里。

注意:保存结果时,需要看情况将已有的结果后移,比如count里是 3,2,2。 这时候统计出一个元素个数是4,如果比较后直接赋值就变成了 4,2,2。需要先将3,2后移一位,最后得到结果4,3,2,a里的元素也要做相应的操作。

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
int cmp(const void *a,const void *b)
{
    return *(int *)a-*(int *)b;
}
int* topKFrequent(int* nums, int numsSize, int k, int* returnSize) {
    qsort(nums,numsSize,sizeof(int),cmp);
    *returnSize=k;
    int i,j,p;
    int *count=(int *)malloc(sizeof(int)*k);
    int *a=(int *)malloc(sizeof(int)*k);
    for(i=0;i<k;i++)
        count[i]=0;
    int tmp=nums[0],index=0,len;
    for(i=0;i<=numsSize;i++)
        if(nums[i]!=tmp||i==numsSize)
        {
            len=i-index;
            for(j=0;j<k;j++)
                if(len>count[j])
                {
                    p=k-1;
                    while(p>j)
                    {
                        count[p]=count[p-1];
                        a[p]=a[p-1];
                        p--;
                    }
                    count[j]=len;
                    a[j]=tmp;
                    break;
                }
            index=i;
            tmp=nums[i];
        }
    return a;
}

 

前K个高频元素

标签:情况   出现   快排   sum   col   size   note   array   sort   

原文地址:https://www.cnblogs.com/onlyandonly/p/9332628.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!