Day93 前 K 個高頻元素

給你一個整數(shù)數(shù)組 nums 和一個整數(shù) k 离例,請你返回其中出現(xiàn)頻率前 k 高的元素。你可以按 任意順序 返回答案

https://leetcode-cn.com/problems/top-k-frequent-elements/

示例1:

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

示例2:

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

提示:

1 <= nums.length <= 105
k 的取值范圍是 [1, 數(shù)組中不相同的元素的個數(shù)]
題目數(shù)據(jù)保證答案唯一度帮,換句話說,數(shù)組中前 k 個高頻元素的集合是唯一的

Java解法

思路:

  • 任意順序很容易想到hash來處理
  • 設(shè)想1:hashmap存儲值以及個數(shù)稿存,最后取出對應(yīng)k大笨篷,過于復(fù)雜
  • 優(yōu)化:參考快排思路,對篩選數(shù)據(jù)進行處理
package sj.shimmer.algorithm.m4_2021;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import sj.shimmer.algorithm.Utils;

/**
 * Created by SJ on 2021/4/30.
 */

class D93 {
    public static void main(String[] args) {
        Utils.logArray(topKFrequent(new int[]{1,1,1,2,2,3},2));
    }
    public static int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            Integer size = map.getOrDefault(num, 0);
            size++;
            map.put(num, size);
        }
        List<int[]> values = new ArrayList<>();
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            int key = entry.getKey();
            int count = entry.getValue();
            values.add(new int[]{key, count});
        }
        int[] result = new int[k];
        qsort(values, 0, values.size() - 1, result, 0, k);
        return result;
    }

    public static void qsort(List<int[]> values, int start, int end, int[] ret, int retIndex, int k) {
        int picked = (int) (Math.random() * (end - start + 1)) + start;
        Collections.swap(values, picked, start);

        int pivot = values.get(start)[1];
        int index = start;
        for (int i = start + 1; i <= end; i++) {
            if (values.get(i)[1] >= pivot) {
                Collections.swap(values, index + 1, i);
                index++;
            }
        }
        Collections.swap(values, start, index);

        if (k <= index - start) {
            qsort(values, start, index - 1, ret, retIndex, k);
        } else {
            for (int i = start; i <= index; i++) {
                ret[retIndex++] = values.get(i)[0];
            }
            if (k > index - start + 1) {
                qsort(values, index + 1, end, ret, retIndex, k - (index - start + 1));
            }
        }
    }
}
image

官方解

https://leetcode-cn.com/problems/top-k-frequent-elements/solution/qian-k-ge-gao-pin-yuan-su-by-leetcode-solution/

  1. 堆:篩選過后使用堆排序處理
  2. 快排:篩選過后使用快排處理(這里我用了官方的快排查找方法)
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子泄伪,更是在濱河造成了極大的恐慌季俩,老刑警劉巖,帶你破解...
    沈念sama閱讀 223,002評論 6 519
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機辜贵,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 95,357評論 3 400
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來归形,“玉大人托慨,你說我怎么就攤上這事∠玖瘢” “怎么了厚棵?”我有些...
    開封第一講書人閱讀 169,787評論 0 365
  • 文/不壞的土叔 我叫張陵,是天一觀的道長蔼紧。 經(jīng)常有香客問我婆硬,道長,這世上最難降的妖魔是什么奸例? 我笑而不...
    開封第一講書人閱讀 60,237評論 1 300
  • 正文 為了忘掉前任彬犯,我火速辦了婚禮,結(jié)果婚禮上查吊,老公的妹妹穿的比我還像新娘谐区。我一直安慰自己,他們只是感情好菩貌,可當我...
    茶點故事閱讀 69,237評論 6 398
  • 文/花漫 我一把揭開白布卢佣。 她就那樣靜靜地躺著,像睡著了一般箭阶。 火紅的嫁衣襯著肌膚如雪虚茶。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,821評論 1 314
  • 那天仇参,我揣著相機與錄音嘹叫,去河邊找鬼。 笑死诈乒,一個胖子當著我的面吹牛罩扇,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播怕磨,決...
    沈念sama閱讀 41,236評論 3 424
  • 文/蒼蘭香墨 我猛地睜開眼喂饥,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了肠鲫?” 一聲冷哼從身側(cè)響起员帮,我...
    開封第一講書人閱讀 40,196評論 0 277
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎导饲,沒想到半個月后捞高,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,716評論 1 320
  • 正文 獨居荒郊野嶺守林人離奇死亡渣锦,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,794評論 3 343
  • 正文 我和宋清朗相戀三年硝岗,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片袋毙。...
    茶點故事閱讀 40,928評論 1 353
  • 序言:一個原本活蹦亂跳的男人離奇死亡型檀,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出听盖,到底是詐尸還是另有隱情贱除,我是刑警寧澤,帶...
    沈念sama閱讀 36,583評論 5 351
  • 正文 年R本政府宣布媳溺,位于F島的核電站月幌,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏悬蔽。R本人自食惡果不足惜扯躺,卻給世界環(huán)境...
    茶點故事閱讀 42,264評論 3 336
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望蝎困。 院中可真熱鬧录语,春花似錦、人聲如沸禾乘。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,755評論 0 25
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽始藕。三九已至蒲稳,卻和暖如春氮趋,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背江耀。 一陣腳步聲響...
    開封第一講書人閱讀 33,869評論 1 274
  • 我被黑心中介騙來泰國打工剩胁, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人祥国。 一個月前我還...
    沈念sama閱讀 49,378評論 3 379
  • 正文 我出身青樓昵观,卻偏偏與公主長得像,于是被迫代替她去往敵國和親舌稀。 傳聞我的和親對象是個殘疾皇子啊犬,可洞房花燭夜當晚...
    茶點故事閱讀 45,937評論 2 361

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