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

 

发布了722 篇原创文章 · 获赞 106 · 访问量 19万+

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/104153996