Не более N повторов

Дан список целых чисел nums и число n. Напишите функцию limit_repeats(nums, n), которая возвращает список с сохранённым порядком, где каждое число встречается не более n раз.

[2, 1, 2, 2, 1, 3, 1, 3], n=2 -> [2, 1, 2, 1, 3, 3].

КАК СПРАШИВАЮТЗадача на один проход со словарём счётчиков. Проверяют, что вы не используете `count` в цикле (O(n²)) и правильно обрабатываете n = 0.

ПОДСКАЗКА

Словарь «число → сколько раз уже взяли»; добавлять в результат, пока счётчик меньше n.

РЕШЕНИЕ И РАЗБОР

Сначала попробуйте сами: на собеседовании подсказки не будет.

Словарь «значение → сколько раз уже взяли». Элемент идёт в результат, пока его счётчик меньше n, счётчик увеличивается в любом случае. Один проход, порядок сохраняется, память O(число различных значений).

def limit_repeats(nums, n):
    seen = {}
    out = []
    for x in nums:
        c = seen.get(x, 0)
        if c < n:
            out.append(x)
        seen[x] = c + 1
    return out

ментор разберёт с вами

песочница · Pythonрешено 0 / 1

Ctrl+Enter

ТЕСТЫ

Ввод	Ваш вывод	Ожидаемый вывод
[1, 2, 3], 1	—	[1, 2, 3]
[2, 1, 2, 2, 1, 3, 1, 3], 2	—	[2, 1, 2, 1, 3, 3]
[1, 1, 1, 1, 1, 1, 1, 1], 3	—	[1, 1, 1]
[5, 5], 0	—	[]
скрытый тест 5

решённые задания сохранятся в прогресс после входа

Не более N повторов

[2, 1, 2, 2, 1, 3, 1, 3], n=2 -> [2, 1, 2, 1, 3, 3].

ПОДСКАЗКА

Словарь «число → сколько раз уже взяли»; добавлять в результат, пока счётчик меньше n.

РЕШЕНИЕ И РАЗБОР

Сначала попробуйте сами: на собеседовании подсказки не будет.

def limit_repeats(nums, n):
    seen = {}
    out = []
    for x in nums:
        c = seen.get(x, 0)
        if c < n:
            out.append(x)
        seen[x] = c + 1
    return out

ментор разберёт с вами

Ввод

Ваш вывод

Ожидаемый вывод

[1, 2, 3], 1

—

[1, 2, 3]

[2, 1, 2, 2, 1, 3, 1, 3], 2

—

[2, 1, 2, 1, 3, 3]

[1, 1, 1, 1, 1, 1, 1, 1], 3

—

[1, 1, 1]

[5, 5], 0

—

[]

скрытый тест 5