【LeetCode】1338.数组大小减半

【LeetCode】1338.数组大小减半

题目描述

给你一个整数数组 arr。你可以从中选出一个整数集合,并删除这些整数在数组中的每次出现。

返回 至少 能删除数组中的一半整数的整数集合的最小大小。
例如:
输入: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],新数组长度大于原数组的二分之一。
再如:
输入:arr = [7,7,7,7,7,7]
输出:1
解释:我们只能选择集合 {7},结果数组为空。

解题思路

首先在语言的选择上,为了方便选了Python(因为Counter类计数很方便)。
然后来分析一下:
一看求最值,考虑往贪心或者dp考虑;由于这题中前后条件划分联系不太大,考虑贪心(DP也是可以的,但是状态转移方程太麻烦。而且这题如果用dp的话空间消耗反而变大了嗯,果然是太懒了)
既然要求求出满足一个最小的和值的状态,那么类似于挑硬币问题,总是贪心地选择最优解即可,因此考虑对数组进行排序,然后根据不大于二分之一这一限制条件进行贪心选择并计数即可

代码

class Solution:
    def minSetSize(self, arr: List[int]) -> int:
        from collections import Counter
        cnt=Counter(arr)
        sort_list = [e[1] for e in sorted(cnt.items(), key = lambda x:-x[1])] #此处的排序要注意边界
        size = (len(arr)+2 - 1)//2 #限制条件
        res=0
        i=0
        temp=0
        while temp < size:
            temp += sort_list[i]
            res += 1 #每满足一个便计数
            i += 1
        return res

总结

总得来说是挑硬币问题PLUS,选用贪心解法在空间开销上有优势。另外务必注意边界条件。

发布了29 篇原创文章 · 获赞 18 · 访问量 3999

猜你喜欢

转载自blog.csdn.net/weixin_44522586/article/details/104485086