当前位置:   article > 正文

leecode题号347:前K个高频元素,python实现_python前k个高频元素 给定一个非空的整数数组,返回其中出现频率前k高的元素。例如

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

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

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解法一:字典排序

  1. class Solution(object):
  2. def topKFrequent(self, nums, k):
  3. """
  4. :type nums: List[int]
  5. :type k: int
  6. :rtype: List[int]
  7. """
  8. dict = {}
  9. for i in nums:
  10. if i not in dict:
  11. dict[i] = 1
  12. else:
  13. dict[i] += 1
  14. print('dict1:')
  15. print(dict)
  16. dict_sort = sorted(dict.items(), key = lambda x: x[1],reverse = True)
  17. print('sorted dict:')
  18. print(dict_sort)
  19. res = []
  20. for i in range(k):
  21. res.append(dict_sort[i][0])
  22. return res

解法二:桶排序

  1. class Solution(object):
  2. def topKFrequent(self, nums, k):
  3. """
  4. :type nums: List[int]
  5. :type k: int
  6. :rtype: List[int]
  7. """
  8. maps = {}
  9. #生成字典映射
  10. for i in nums:
  11. # print('i:%d,maps.get(i,0):%d' %(i, maps.get(i,0)))
  12. maps[i] = maps.get(i,0) + 1
  13. max_time = max(maps.values())
  14. print(maps)
  15. #根据最大次数生成桶
  16. TongList = [[] for i in range(max_time+1)]
  17. for key ,value in maps.items():
  18. TongList[value].append(key)
  19. print(TongList)
  20. res = []
  21. for i in range(max_time, 0, -1):
  22. print('i:%d' %(i))
  23. if TongList[i]:
  24. res.extend(TongList[i])
  25. if len(res) >= k:
  26. return res[:k]

 

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号