Enable Javascript in your browser and then refresh this page, for a much enhanced experience.
First solution in Uncategorized category for When "k" is Enough! by saitama-1
from typing import Iterable
def remove_after_kth(items: list, k: int) -> Iterable:
dict = {}
lst = []
for item in items:
if item not in dict:
dict[item] = 1
else:
dict[item] += 1
if dict[item] <= k:
lst.append(item)
return lst
print("Example:")
print(list(remove_after_kth([42, 42, 42, 42, 42, 42, 42], 3)))
assert list(remove_after_kth([42, 42, 42, 42, 42, 42, 42], 3)) == [42, 42, 42]
assert list(remove_after_kth([42, 42, 42, 99, 99, 17], 0)) == []
assert list(remove_after_kth([1, 1, 1, 2, 2, 2], 5)) == [1, 1, 1, 2, 2, 2]
print("The mission is done! Click 'Check Solution' to earn rewards!")
Nov. 14, 2022