1. 程式人生 > >前K個高頻元素

前K個高頻元素

給定 assume clas 其中 span 復雜 top != int

給定一個非空的整數數組,返回其中出現頻率前 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; }

前K個高頻元素