LeetCode-347:Top K Frequent Elements(取前k个频率最高的元素)

题目:

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

例子:

Example:
Given [1,1,1,2,2,3] and k = 2, return [1,2].
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个频率最高的元素。
链接:LeetCode:https://leetcode.com/problems/top-k-frequent-elements/description/

解题思路:

使用哈希表(python 中字典)来存储各个元素出现的次数,键(key)为对应元素,值(value)为对应元素出现个数。再对该哈希表(字典)按照值进行排序。
代码(python):

class Solution(object):
    def topKFrequent(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: List[int]
        """
        Dict = {}
        for i in range(len(nums)):
            if nums[i] in Dict:
                Dict[nums[i]] = Dict[nums[i]] + 1
            else:
                Dict[nums[i]] = 1

        output = sorted(Dict.items(), key=lambda e: e[1], reverse=True)

        final = []
        for i in range(k):
            final.append(output[i][0])

        return final

solution = Solution()

output = solution.topKFrequent([1,1,1,2,2,3],2)

print(output)

关于程序中python内置函数sorted方法的备注

  • 描述
    • sorted() 函数对所有可迭代的对象进行排序操作。
  • sort 与 sorted 区别:

    • sort 是应用在 list 上的方法,sorted 可以对所有可迭代的对象进行排序操作。
    • list 的 sort 方法返回的是对已经存在的列表进行操作,而内建函数 sorted 方法返回的是一个新的 list,而不是在原来的基础上进行的操作。
  • 语法

    • sorted 语法:
      sorted(iterable[, cmp[, key[, reverse]]])
    • 参数说明:
      • iterable – 可迭代对象。
      • cmp – 比较的函数,这个具有两个参数,参数的值都是从可迭代对象中取出,此函数必须遵守的规则为,大于则返回1,小于则返回-1,等于则返回0。
      • key – 主要是用来进行比较的元素,只有一个参数,具体的函数的参数就是取自于可迭代对象中,指定可迭代对象中的一个元素来进行排序。
      • reverse – 排序规则,reverse = True 降序 , reverse = False 升序(默认)。
  • 返回值
    返回重新排序的列表。
  • 实例
    以下实例展示了 sorted 的使用方法
>>>a = [5,7,6,3,4,1,2]
>>> b = sorted(a)       # 保留原列表
>>> a 
[5, 7, 6, 3, 4, 1, 2]
>>> b
[1, 2, 3, 4, 5, 6, 7]

>>> L=[('b',2),('a',1),('c',3),('d',4)]
>>> sorted(L, cmp=lambda x,y:cmp(x[1],y[1]))   # 利用cmp函数
[('a', 1), ('b', 2), ('c', 3), ('d', 4)]
>>> sorted(L, key=lambda x:x[1])               # 利用key
[('a', 1), ('b', 2), ('c', 3), ('d', 4)]

>>> students = [('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
>>> sorted(students, key=lambda s: s[2])            # 按年龄排序
[('dave', 'B', 10), ('jane', 'B', 12), ('john', 'A', 15)]

>>> sorted(students, key=lambda s: s[2], reverse=True)       # 按降序
[('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
>>>
参考网址:
  • [1]: http://www.runoob.com/python/python-func-sorted.html
  • [2]: https://blog.csdn.net/xiaoxiaoley/article/details/78952751
  • [3]: https://blog.csdn.net/koala_tree/article/details/80393689

你可能感兴趣的:(编程算法)