Задача на перекладывание кубиков

@russian_hustla · Регистрация: 30.10.2017

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

Гена очень послушный и умный маленький мальчик. Как и все дети он любит играть в кубики, но часто ему в голову приходят сложные математические формули или непростые задачи. Вот и в этот раз, когда он раскладывал кубики, то придумал очень интересную задачу.

Он начал выкладывать кубики в ряд длинной n. Причем высота на i-ом месте ряда составляет a[i] кубиков. Тут он и задался вопросом: какое минимальное количество кубиков надо переложить, чтобы все высоты стали равны. Кубики можно перекладывать только на соседние места, то есть, если мы взяли кубик с ряда номер j, то можно положить его на ряд номер j+1 или j−1, если такие существуют.

Формат входных данных
Первая строка содержит одно целое число nn (1≤n≤10^5) - количество рядов из кубиков, которые выложил Гена.

Во второй строке дано nn чисел aiai(0≤a[i]≤10^6) - количество кубиков в i-м номере ряда.

Формат выходных данных
Выведите минимально количество перекладываний, которое необходимо сделать Гене, чтобы все высоты стали равны, если Гена не сможет выстроить такой ряд из кубиков, выведите −1.

Пример ввода 1:
4
2 1 2 3

Пример вывода 2:
2

Пример ввода 1:
5
2 3 4 5 6

Пример вывода 2:
10

Я написал код, но он проходит не все тесты:

Python
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
n = int(input())
s1 = input()
s1 = s1.split()
t = 0
 
for i in range(n):
    s1[i] = int(s1[i])
 
if sum(s1) % n == 0:
    d = int(sum(s1) / n)
    while s1.count(d) < n:
        imax = max(s1)
        ind = s1.index(imax)
        if ind != (n - 1):
            if ((s1[ind - 1] < imax) or (s1[ind + 1] < imax)):
                s1[ind] -= 1
                m = min(s1[ind - 1],s1[ind + 1])
                if m == s1[ind - 1]:
                    mind = ind - 1
                else: mind = ind + 1
                s1[mind] += 1
                t += 1
        else:
            if (s1[ind - 1] < imax):
                s1[ind] -= 1
                s1[ind - 1] += 1
                t += 1
    print(t)
else:
    print('-1')

@Кот_Василий · 31.10.2017, 07:28

Ты решил?

Добавлено через 34 секунды
Есть вк ?

@russian_hustla · 31.10.2017, 11:59 **[ТС]**

Нет, не решил

Добавлено через 44 секунды
Кот_Василий, https://vk.com/russian_hustla

@Дмитрий Власов · 31.10.2017, 13:15

у вас программа по времени не проходит. оптимизируйте

Добавлено через 8 минут
потребуется дополнительно одна (или две) функция
и используйте рекурсию. Удачи в олимпиаде НТИ !!!

Новые блоги и статьи Все статьи Все блоги /
http://iceja.net/ математические сервисы iceja 20.01.2026 Обновила свой сайт http:/ / iceja. net/ , приделала Fast Fourier Transform экстраполяцию сигналов. Однако предсказывает далеко не каждый сигнал (см ограничения http:/ / iceja. net/ fourier/ docs ). Также. . .	http://iceja.net/ сервер решения полиномов iceja 18.01.2026 Выкатила http:/ / iceja. net/ сервер решения полиномов (находит действительные корни полиномов методом Штурма). На сайте документация по API, но скажу прямо VPS слабенький и 200 000 полиномов. . .	Расчёт переходных процессов в цепи постоянного тока igorrr37 16.01.2026 / * Дана цепь постоянного тока с R, L, C, k(ключ), U, E, J. Программа составляет систему уравнений по 1 и 2 законам Кирхгофа, решает её и находит переходные токи и напряжения на элементах схемы. . . .	Восстановить юзерскрипты Greasemonkey из бэкапа браузера damix 15.01.2026 Если восстановить из бэкапа профиль Firefox после переустановки винды, то список юзерскриптов в Greasemonkey будет пустым. Но восстановить их можно так. Для этого понадобится консольная утилита. . .
Сукцессия микоризы: основная теория в виде двух уравнений. anaschu 11.01.2026 https:/ / rutube. ru/ video/ 7a537f578d808e67a3c6fd818a44a5c4/	WordPad для Windows 11 Jel 10.01.2026 WordPad для Windows 11 — это приложение, которое восстанавливает классический текстовый редактор WordPad в операционной системе Windows 11. После того как Microsoft исключила WordPad из. . .	Classic Notepad for Windows 11 Jel 10.01.2026 Old Classic Notepad for Windows 11 Приложение для Windows 11, позволяющее пользователям вернуть классическую версию текстового редактора «Блокнот» из Windows 10. Программа предоставляет более. . .	Почему дизайн решает? Neotwalker 09.01.2026 В современном мире, где конкуренция за внимание потребителя достигла пика, дизайн становится мощным инструментом для успеха бренда. Это не просто красивый внешний вид продукта или сайта — это. . .

@Кот_Василий 0 / 0 / 0 Регистрация: 31.10.2017 Сообщений: 1
	31.10.2017, 07:28
	Ты решил? Добавлено через 34 секунды Есть вк ? 0

@russian_hustla 0 / 0 / 0 Регистрация: 30.10.2017 Сообщений: 2
	31.10.2017, 11:59 [ТС]
	Нет, не решил Добавлено через 44 секунды Кот_Василий, https://vk.com/russian_hustla 0

@Дмитрий Власов 1 / 1 / 0 Регистрация: 10.11.2016 Сообщений: 1
	31.10.2017, 13:15
	у вас программа по времени не проходит. оптимизируйте Добавлено через 8 минут потребуется дополнительно одна (или две) функция и используйте рекурсию. Удачи в олимпиаде НТИ !!! 1