LeetCode-Python-1342. 数组大小减半(贪心 + 堆)
作者:互联网
给你一个整数数组 arr。你可以从中选出一个整数集合,并删除这些整数在数组中的每次出现。
返回 至少 能删除数组中的一半整数的整数集合的最小大小。
示例 1:
输入:arr = [3,3,3,3,5,5,5,2,2,7]
输出:2
解释:选择 {3,7} 使得结果数组为 [5,5,5,2,2]、长度为 5(原数组长度的一半)。
大小为 2 的可行集合有 {3,5},{3,2},{5,2}。
选择 {2,7} 是不可行的,它的结果数组为 [3,3,3,3,5,5,5],新数组长度大于原数组的二分之一。
示例 2:
输入:arr = [7,7,7,7,7,7]
输出:1
解释:我们只能选择集合 {7},结果数组为空。
示例 3:
输入:arr = [1,9]
输出:1
示例 4:
输入:arr = [1000,1000,3,7]
输出:1
示例 5:
输入:arr = [1,2,3,4,5,6,7,8,9,10]
输出:5
提示:
1 <= arr.length <= 10^5
arr.length 为偶数
1 <= arr[i] <= 10^5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/reduce-array-size-to-the-half
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
根据贪心的思想,每次应当删当前次数最多的元素,这样使得数组大小减小的幅度最大。
所以不妨先统计每个元素出现的频率,然后从出现频率最大的元素开始删起。
找最大值可以利用最大堆。
时间复杂度:O(NlogN)
空间复杂度:O(N)
from heapq import *
from collections import Counter
class Solution(object):
def minSetSize(self, arr):
"""
:type arr: List[int]
:rtype: int
"""
t = len(arr) // 2
dic = Counter(arr)
queue = []
for key, val in dic.items():
heappush(queue, -val)
cnt = 0
res = 0
while cnt < t:
tmp = heappop(queue)
res += 1
cnt += -tmp
# print cnt, tmp, t
return res
![](https://g.csdnimg.cn/static/user-reg-year/1x/4.png)
标签:arr,示例,Python,res,1342,cnt,整数,数组,LeetCode 来源: https://blog.csdn.net/qq_32424059/article/details/104153996