前 K 個高頻元素

題目來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/top-k-frequent-elements

給定一個非空的整數(shù)數(shù)組垛膝,返回其中出現(xiàn)頻率前 k 高的元素柏副。

示例 1:

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

示例 2:

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

說明:

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

利用優(yōu)先隊列(小根堆)的解法:

class Solution {

    class Node{
        int count;
        int key;
        public Node(int count,int key){
            this.count = count;
            this.key = key;
        }
    }

    public List<Integer> topKFrequent(int[] nums, int k) {
        Map<Integer,Integer> map = new HashMap<>();
        for(int num : nums){
            Integer count = map.get(num);
            if (count == null){
                map.put(num,1);
            }else {
                map.put(num,count+1);
            }
        }
        PriorityQueue<Node> queue = new PriorityQueue<>((n1,n2)->n1.count-n2.count);
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            queue.add(new Node(entry.getValue(),entry.getKey()));
            if (queue.size() > k)
                queue.poll();
        }
        List<Integer> ans = new ArrayList<>();
        while(!queue.isEmpty()){
            Node node = queue.poll();
            ans.add(node.key);
        }
        return ans;
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌骚勘,老刑警劉巖,帶你破解...
    沈念sama閱讀 223,126評論 6 520
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件撮奏,死亡現(xiàn)場離奇詭異俏讹,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)畜吊,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,421評論 3 400
  • 文/潘曉璐 我一進(jìn)店門泽疆,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人玲献,你說我怎么就攤上這事殉疼。” “怎么了捌年?”我有些...
    開封第一講書人閱讀 169,941評論 0 366
  • 文/不壞的土叔 我叫張陵瓢娜,是天一觀的道長。 經(jīng)常有香客問我礼预,道長眠砾,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 60,294評論 1 300
  • 正文 為了忘掉前任托酸,我火速辦了婚禮褒颈,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘励堡。我一直安慰自己谷丸,他們只是感情好,可當(dāng)我...
    茶點故事閱讀 69,295評論 6 398
  • 文/花漫 我一把揭開白布念秧。 她就那樣靜靜地躺著淤井,像睡著了一般布疼。 火紅的嫁衣襯著肌膚如雪摊趾。 梳的紋絲不亂的頭發(fā)上币狠,一...
    開封第一講書人閱讀 52,874評論 1 314
  • 那天,我揣著相機(jī)與錄音砾层,去河邊找鬼漩绵。 笑死,一個胖子當(dāng)著我的面吹牛肛炮,可吹牛的內(nèi)容都是我干的止吐。 我是一名探鬼主播,決...
    沈念sama閱讀 41,285評論 3 424
  • 文/蒼蘭香墨 我猛地睜開眼侨糟,長吁一口氣:“原來是場噩夢啊……” “哼碍扔!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起秕重,我...
    開封第一講書人閱讀 40,249評論 0 277
  • 序言:老撾萬榮一對情侶失蹤不同,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后溶耘,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體二拐,經(jīng)...
    沈念sama閱讀 46,760評論 1 321
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,840評論 3 343
  • 正文 我和宋清朗相戀三年凳兵,在試婚紗的時候發(fā)現(xiàn)自己被綠了百新。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 40,973評論 1 354
  • 序言:一個原本活蹦亂跳的男人離奇死亡庐扫,死狀恐怖饭望,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情形庭,我是刑警寧澤杰妓,帶...
    沈念sama閱讀 36,631評論 5 351
  • 正文 年R本政府宣布,位于F島的核電站碘勉,受9級特大地震影響巷挥,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜验靡,卻給世界環(huán)境...
    茶點故事閱讀 42,315評論 3 336
  • 文/蒙蒙 一倍宾、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧胜嗓,春花似錦高职、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,797評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春埃元,著一層夾襖步出監(jiān)牢的瞬間涝涤,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,926評論 1 275
  • 我被黑心中介騙來泰國打工岛杀, 沒想到剛下飛機(jī)就差點兒被人妖公主榨干…… 1. 我叫王不留阔拳,地道東北人。 一個月前我還...
    沈念sama閱讀 49,431評論 3 379
  • 正文 我出身青樓类嗤,卻偏偏與公主長得像糊肠,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子遗锣,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 45,982評論 2 361

推薦閱讀更多精彩內(nèi)容