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

@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 формируется из целых чисел.

Новые блоги и статьи Все статьи Все блоги /
Фото всей Земли с борта корабля Orion миссии Artemis II kumehtar 04.04.2026 Это первое подобное фото сделанное человеком за 50 лет. Снимок называют новым вариантом легендарной фотографии «The Blue Marble» 1972 года, сделанной с борта корабля «Аполлон-17». Новое фото. . .	Вывод диалогового окна перед закрытием, если документ не проведён Maks 04.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: реализовать программный контроль на предмет проведения документа. . .	Программный контроль заполнения реквизита табличной части документа Maks 02.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: реализовать контроль заполнения реквизита "ПричинаСписания". . .	wmic не является внутренней или внешней командой Maks 02.04.2026 Решение: DISM / Online / Add-Capability / CapabilityName:WMIC~~~~ Отсюда: https:/ / winitpro. ru/ index. php/ 2025/ 02/ 14/ komanda-wmic-ne-naydena/
Программная установка даты и запрет ее изменения Maks 02.04.2026 Алгоритм из решения ниже реализован на примере нетипового документа "СписаниеМатериалов", разработанного в конфигурации КА2. Задача: при создании документов установить период списания автоматически. . .	Вывод данных в справочнике через динамический список Maks 01.04.2026 Реализация из решения ниже выполнена на примере нетипового справочника "Спецтехника" разработанного в конфигурации КА2. Задача: вывести данные из ТЧ нетипового документа. . .	Программное заполнения текстового поля в реквизите формы документа Maks 01.04.2026 Алгоритм из решения ниже реализован на нетиповом документе "ВыдачаОборудованияНаСпецтехнику" разработанного в конфигурации КА2, в дополнении к предыдущему решению. На форме документа создается. . .	К слову об оптимизации kumehtar 01.04.2026 Вспоминаю начало 2000-х, университет, когда я писал на Delphi. Тогда среди программистов на форумах активно обсуждали аккуратную работу с памятью: нужно было следить за переменными, вовремя. . .