删除k个元素后的最大不同元素

如何解决删除k个元素后的最大不同元素

这是geeksforgeeks问题的链接 https://practice.geeksforgeeks.org/problems/maximum-distinct-elements-after-removing-k-elements/0

这是我的代码-

import collections
import heapq
def maxoccur(arr,k):
   c = 0
   d = collections.Counter(arr)
   heap = [(value,key) for key,value in d.items()]
   heapq.heapify(heap)
   while(heap):
     x = heapq.heappop(heap)

     if x[0] == 1:
        c+=1
     elif x[0]>1 and k>0:
        k-=1
        y = x[0]-1
        heapq.heappush(heap,(y,x[1]))
     elif k<=0:
        break
    if k == 0:
        while(heap):
            x = heapq.heappop(heap)
            if x[0] == 1:
               c+=1
    elif k>0:
       c-=k
    
    return c

我的代码有什么问题我在这个测试案例中得到了错误的答案-

输入： 84 47 28 26 24 26 17 13 10 2 3 8 21 20 24 17 1 7 23 17 12 9 28 10 3 21 3 14 8 26 30 13 13 19 19 30 28 14 17 2 23 10 4 22 30 15 8 9 15 6 1 24 17 2 21 27 4 3 21 17 2 16 16 15 28 27 6 17 10 14 18 25 16 13 16 15 28 15 15 4 21 8 19 7 9 9 25

其正确的输出是： 27

您的代码的输出为： 25

解决方法

更简单的方法基于这样的事实：我们首先可以删除出现次数> 1的所有项目，而无需更改不同值的数量，然后删除其余一些项目-现在，不同值的数量会减少。

def maxoccur(arr,k):
    lena = len(arr)
    lens = len(set(arr))
    excess = lena - lens
    return lens if excess >= k else max(0,lena - k)

---

旧答案是作者代码的转换

请注意，堆是minheap，我通过对计数器的求反使它成为准maxheap。 然后，我们仅进行k个（如果可能的话）减少计数器的操作，然后在堆中查找其余的内容。

堆中根本不需要源值，但是我懒得更改代码。

import collections
import heapq
def maxoccur(arr,k):
    d = collections.Counter(arr)
    heap = [(-value,key) for key,value in d.items()]
    heapq.heapify(heap)
    while (heap) and (k > 0):
        k -= 1
        x = heapq.heappop(heap)
        if x[0] < -1:
            heapq.heappush(heap,(x[0]+1,x[1]))
    return len(heap)

k = 47
arr = [int(i) for i in '28 26 24 26 17 13 10 2 3 8 21 20 24 17 1 7 23 17 12 9 28 10 3 21 3 14 8 26 30 13 13 19 30 28 14 17 2 23 10 4 22 30 15 8 9 15 6 1 24 17 2 21 27 4 3 21 17 2 16 16 15 28 27 6 17 10 14 18 25 16 13 16 15 28 15 15 4 21 8 19 7 9 9 25'.split()]
print(maxoccur(arr,k))

arr = [5,7,5,1,2,2]
k = 3
print(maxoccur(arr,k))

>>>
27
4