前K個高頻元素
阿新 • • 發佈:2018-07-19
給定 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個高頻元素