347. 前 K 個高頻元素

給定一個非空的整數數組，返回其中出現頻率前 k 高的元素。

示例 1:

輸入: nums = [1,1,1,2,2,3], k = 2
輸出: [1,2]
示例 2:

輸入: nums = [1], k = 1
輸出: [1]
說明：

你可以假設給定的 k 總是合理的，且 1 ≤ k ≤ 數組中不相同的元素的個數。
你的算法的時間復雜度必須優于 O(n log n) , n 是數組的大小。

class Solution {//前 K 個高頻元素public List<Integer> topKFrequent(int[] nums, int k) {List<Integer> list=new ArrayList();Arrays.sort(nums);int distinctLen=1;for(int i=1;i<nums.length;i++){if(nums[i]!=nums[i-1]){distinctLen++;}}int counts[]=new int[distinctLen];int order[]=new int[distinctLen];int index=0;int count=1;for(int i=1;i<nums.length;i++){if(nums[i]==nums[i-1]){count++;}else{counts[index]=count;order[index]=count;nums[index]=nums[i-1];index++;count=1;}}nums[index]=nums[nums.length-1];counts[index]=count;order[index]=count;Arrays.sort(order);int kth=order[distinctLen-k];for(int i=0;i<=index;i++){if(counts[i]>=kth){list.add(nums[i]);}}return list;} }

總結

以上是生活随笔為你收集整理的Java实现 LeetCode 347 前 K 个高频元素的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。