Разбейте массив на минимально возможное количество отрезков

@pluz0re · Регистрация: 27.03.2022

Студворк — интернет-сервис помощи студентам

Задача следующая:

Ограничение времени: 2 секунды
Ограничение памяти: 256Mb
Дан массив целых положительных чисел a длины n. Разбейте его на минимально возможное количество отрезков, чтобы каждое число было не меньше длины отрезка которому оно принадлежит. Длиной отрезка считается количество чисел в нем.
Разбиение массива на отрезки считается корректным, если каждый элемент принадлежит ровно одному отрезку.
Формат ввода:
Первая строка содержит одно целое число t (1 ≤ t ≤ 1 000) — количество наборов тестовых данных. Затем следуют t наборов тестовых данных.
Первая строка набора тестовых данных содержит одно целое число n (1 ≤ n ≤ 105) — длину массива.
Следующая строка содержит n целых чисел a1, a2, …, an (1 ≤ ai ≤ n) — массив a.
Гарантируется, что сумма n по всем наборам тестовых данных не превосходит 2 ⋅ 10**5.
Формат вывода:
Для каждого набора тестовых данных в первой строке выведите число k — количество отрезков в вашем разбиении.
Затем в следующей строке выведите k чисел len1, len2, …, lenk — длины отрезков в порядке слева направо.

Вот мой код:

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
t = int(input())
for test in range(t):
    n = int(input())
    k, test, i = [], [], 0
    a = [int(i) for i in input().split()]
    while i < n:
        if int(min(test, default=1)) >= len(test)+1 and int(a[i]) >= len(test)+1:
            test.append(a[i])
            i += 1
        else:
            k.append(test)
            test = []
    k.append(test)
    print(len(k))
    [print(len(x), end = ' ') for x in k]

Вопрос:
Тестирующая среда на последнем месте пишет временной лимит превышен. Требуется 2 сек, у меня 2.8 сек. Помогите пожалуйста оптимизировать программу чтобы сократить временной разрыв. Переписывать алгоритм или есть мельчайшие помарки?

@Treasure Hunter · 28.03.2022, 21:41

Особо не поможет, но можно оба int убрать здесь:

Python
1
if int(min(test, default=1)) >= len(test)+1 and int(a[i]) >= len(test)+1:

Смысла в них нет, так как перед этим список a формируется из целых чисел.

Новые блоги и статьи Все статьи Все блоги /
Подключение Box2D v3 к SDL3 для Android: физика и отрисовка коллайдеров 8Observer8 29.01.2026 Содержание блога Box2D - это библиотека для 2D физики для анимаций и игр. С её помощью можно определять были ли коллизии между конкретными объектами. Версия v3 была полностью переписана на Си, в. . .	Инструменты COM: Сохранение данный из VARIANT в файл и загрузка из файла в VARIANT bedvit 28.01.2026 Сохранение базовых типов COM и массивов (одномерных или двухмерных) любой вложенности (деревья) в файл, с возможностью выбора алгоритмов сжатия и шифрования. Часть библиотеки BedvitCOM Использованы. . .	Загрузка PNG с альфа-каналом на SDL3 для Android: с помощью SDL_LoadPNG (без SDL3_image) 8Observer8 28.01.2026 Содержание блога SDL3 имеет собственные средства для загрузки и отображения PNG-файлов с альфа-каналом и базовой работы с ними. В этой инструкции используется функция SDL_LoadPNG(), которая. . .	Загрузка PNG с альфа-каналом на SDL3 для Android: с помощью SDL3_image 8Observer8 27.01.2026 Содержание блога SDL3_image - это библиотека для загрузки и работы с изображениями. Эта пошаговая инструкция покажет, как загрузить и вывести на экран смартфона картинку с альфа-каналом, то есть с. . .
Влияние грибов на сукцессию anaschu 26.01.2026 Бифуркационные изменения массы гриба происходят тогда, когда мы уменьшаем массу компоста в 10 раз, а скорость прироста биомассы уменьшаем в три раза. Скорость прироста биомассы может уменьшаться за. . .	Воспроизведение звукового файла с помощью SDL3_mixer при касании экрана Android 8Observer8 26.01.2026 Содержание блога SDL3_mixer - это библиотека я для воспроизведения аудио. В отличие от инструкции по добавлению текста код по проигрыванию звука уже содержится в шаблоне примера. Нужно только. . .	Установка Android SDK, NDK, JDK, CMake и т.д. 8Observer8 25.01.2026 Содержание блога Перейдите по ссылке: https:/ / developer. android. com/ studio и в самом низу страницы кликните по архиву "commandlinetools-win-xxxxxx_latest. zip" Извлеките архив и вы увидите. . .	Вывод текста со шрифтом TTF на Android с помощью библиотеки SDL3_ttf 8Observer8 25.01.2026 Содержание блога Если у вас не установлены Android SDK, NDK, JDK, и т. д. то сделайте это по следующей инструкции: Установка Android SDK, NDK, JDK, CMake и т. д. Сборка примера Скачайте. . .