LeetCode 347. Top K Frequent Elements

Description

Given a non-empty array of integers, return the k most frequent elements.

Example 1:

Input: nums = [1,1,1,2,2,3], k = 2
Output: [1,2]
Example 2:

Input: nums = [1], k = 1
Output: [1]
Note:

You may assume k is always valid, 1 ≤ k ≤ number of unique elements.
Your algorithm's time complexity must be better than O(n log n), where n is the array's size.

描述

給定一個非空的整數數組，返回其中出現頻率前 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 是數組的大小。

思路

• 使用字典，統計每一個元素出現的次數，元素爲鍵，元素出現的次數爲值。
• 而後以元素出現的次數爲值，統計該次數下出現的全部的元素。
• 從最大次數遍歷到 1 次，若該次數下有元素出現，提取該次數下的全部元素到結果數組中，知道提取到 k 個元素爲止。

# -*- coding: utf-8 -*-
# @Author:             何睿
# @Create Date:        2019-04-09 12:37:36
# @Last Modified by:   何睿
# @Last Modified time: 2019-04-09 15:57:00

from collections import Counter


class Solution:
    def topKFrequent(self, nums: [int], k: int) -> [int]:
        # 桶
        bucket = dict()
        # 構建字典，鍵位數字，值爲該數字出現過的次數
        table = Counter(nums)
        result, count = [], 0
        # 以元素出現的次數位鍵，該次數下的全部元素構成的 List 爲值
        for num, times in table.items():
            if times not in bucket: bucket[times] = []
            bucket[times].append(num)
        # 出現的最大次數
        maxtime = max(table.values())

        for time in range(maxtime, 0, -1):
            # 若是該次數下有元素
            if time in bucket:
                # 提取當前次數下的全部元素到結果中
                result.extend(bucket[time])
                count += len(bucket[time])
            if count == k: break
        return result

源代碼文件在 這裏 。
©本文首發於 何睿的博客 ，歡迎轉載，轉載需保留 文章來源 ，做者信息和本聲明.