Назад к вопросам
Junior
23
Как решить задачу за линейное время O(n), используя словарь? Что нужно хранить в словаре, чтобы потом собрать отсортированную строку? Как обработать символы, которых нет в order?
Компании, где спрашивали
Яндекс
Ответ от нейросети
sobes.tech AI
Чтобы решить задачу за линейное время O(n) с помощью словаря (хэша), нужно:
-
Подсчитать количество каждого символа в исходной строке и сохранить в словаре, где ключ — символ, значение — количество.
-
Итерироваться по строке
order, для каждого символа брать из словаря его количество и добавлять этот символ столько раз в результат. -
Обработать символы, которых нет в
order— пройтись по словарю и добавить оставшиеся символы в любом порядке (например, в порядке появления).
Пример на Python:
from collections import Counter
def custom_sort(s: str, order: str) -> str:
count = Counter(s)
result = []
for ch in order:
if ch in count:
result.append(ch * count[ch])
del count[ch]
# Добавляем символы, которых нет в order
for ch, cnt in count.items():
result.append(ch * cnt)
return ''.join(result)
Такой подход гарантирует, что каждый символ обрабатывается один раз, что обеспечивает сложность O(n).