前K个高频元素

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

例如,

给定数组 [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;
}

 

posted @ 2018-07-18 21:11  onlyandonly  阅读(176)  评论(0编辑  收藏  举报