【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,选用贪心解法在空间开销上有优势。另外务必注意边界条件。