Leetcode 1338.数组大小减半
1 题目描述(Leetcode题目链接)
给你一个整数数组 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],新数组长度大于原数组的二分之一。
2 题解
用哈希表和排序来做,首先统计每个字出现的次数,然后按照次数逆序排序,用长度逐一减去各个数字的出现次数直到长度小于等于原先的一半。
class Solution:
def minSetSize(self, arr: List[int]) -> int:
dic = {}
length = len(arr)
half = length//2
retv = 0
for num in arr:
if num not in dic:
dic[num] = 1
else:
dic[num] += 1
sortdic = sorted(dic.items(), key = lambda x:x[1], reverse = True)
for key, value in sortdic:
length -= value
retv += 1
if length <= half:
return retv
评论区的神仙调各种库用一行代码就实现了。
return bisect.bisect_left(list(itertools.accumulate(sorted(collections.Counter(arr).values(), reverse=True))), len(arr) // 2) + 1