前K个高频元素

Posted 2020-12-05 onlyandonly

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

例如，

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

注意：

• 你可以假设给定的 k 总是合理的，1 ≤ k ≤ 数组中不相同的元素的个数。
• 你的算法的时间复杂度必须优于 O(n log n) , 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;
}