堆相關 347 前 K 個高頻元素

2021-10-10 06:49:27 字數 699 閱讀 3536

給定乙個非空的整數陣列,返回其**現頻率前 k 高的元素。

示例 1:

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

輸出: [1,2]

示例 2:

輸入: nums = [1], k = 1

輸出: [1]

在這裡,我們可以利用堆的思想:建立乙個小頂堆,然後遍歷「出現次數陣列」:

如果堆的元素個數小於 kk,就可以直接插入堆中。

如果堆的元素個數等於 kk,則檢查堆頂與當前出現次數的大小。如果堆頂更大,說明至少有 kk 個數字的出現次數比當前值大,故捨棄當前值;否則,就彈出堆頂,並將當前值插入堆中。

遍歷完成後,堆中的元素就代表了「出現次數陣列」中前 kk 大的值

class solution 

// int 的第乙個元素代表陣列的值,第二個元素代表了該值出現的次數

priorityqueuequeue = new priorityqueue(new comparator()

});for (map.entryentry : occurrences.entryset()) );

}} else );}}

int ret = new int[k];

for (int i = 0; i < k; ++i)

return ret;

}}

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...

347 前 K 個高頻元素

維護乙個size是k的堆。求前k大的數,用小根堆 求前k小的數,用大根堆 求前k大,當堆的sizek時,去掉堆頂,最終堆中的資料就是最後的答案。如陣列arr 求最大的前3個數 一般來說求前k大 小,陣列中數是很大的,直接用排序,在選擇前k大 小的話,效率不高 首先建立堆,其size 0,一直push...

347 前 K 個高頻元素

方法很好想,要麼暴力排序,歸併,堆排都能滿足o nlogn 的要求。我第一次想到的就是,用雜湊統計一下數字,然後過載排一下序,順序放入集合裡就行。這次用的是最小堆,主要學習一下priorityqueue,是基於優先順序堆的無界優先順序queue modifier and type method an...