leetcode-347-前K个高频元素

题目描述

给定一个非空的整数数组,返回其中出现频率前 k 高的元素。

提示:

你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
题目数据保证答案唯一,换句话说,数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案。

示例


  • 解题思路

    1. python对字典的操作要熟练.
    2. 基本思路是先把元素和元素个数存入字典, 然后反转key-value.
    3. 因为value 有重复的情况, 所以把重复对应的key以List形式作为值.
    4. 再次就是对字典按键排序, 值列表合并, 列表二维转一维.
  • 题解1:

    执行用时:44 ms, 在所有 Python3 提交中击败了94.56%的用户

    内存消耗:16.4 MB, 在所有 Python3 提交中击败了93.97%的用户

    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    from typing import List, Dict, Iterable
    import operator
    from functools import reduce

    class Solution:
    def topKFrequent(self, nums: List[int], k: int) -> List[int]:
    rd = {} # type: Dict[int, int]
    cd = {} # type: Dict[int, List[int]]
    # 把元素和元素个数对存入字典
    for n in nums:
    if n not in rd:
    rd[n] = 1
    else:
    rd[n] = rd[n] + 1
    # 反转key-value
    for x in rd:
    if rd[x] not in cd:
    cd[rd[x]] = [x]
    else:
    cd.setdefault(rd[x], []).append(x)
    # 字典按键排序
    cd_reverse = sorted(cd.keys(), reverse=True)
    # 值(列表)合并
    rl = [cd[x] for x in cd_reverse] # type: List[List[int]]
    # 二维数组转一维
    result = reduce(operator.add, rl) #type: List[int]
    return result[0:k]
文章作者: Spaceack
文章链接: http://spaceack.com/2020/09/07/2020-09-07-leetcode-347-%E5%89%8DK%E4%B8%AA%E9%AB%98%E9%A2%91%E5%85%83%E7%B4%A0/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 丸子家的小云吞
支付宝打赏
微信打赏