Mistrzostwa Polski Szkół Średnich w Programowaniu Zespołowym 2024
Problem description
Maciej jest koneserem frytek belgijskich. Ciężko dogodzić jego wymaganiom, z tego powodu co roku, w każdą jesień, samodzielnie wybiera się na pole ziemniaków, żeby wybrać ziemniaki najlepszej jakości. Pole składa się z jednej, długiej grządki, w której rośnie N ziemniaków, jeden za drugim. Maciej każdemu ziemniakowi przypisuje pewien stopień okazałości Ai. Oczywiście im ziemniak jest bardziej okazały, tym lepiej.
Maciejowi do zapasów na następny rok potrzebne jest dokładnie K ziemniaków. Zależy mu na tym, żeby sumaryczny stopień okazałości ziemniaków był jak największy. Z drugiej strony zdaje sobie sprawę, że ziemniaki, które rosną w dużej odległości od siebie, będą się od siebie znacząco różniły. Stąd dla danego podciągu ziemniaków (Ai1,…,AiK), gdzie i1 < i2 < …iK, jego jakość Maciej oblicza jako sumę okazałości ziemniaków minus różnicę odległości dwóch najbardziej odległych ziemniaków, tj.:
$$(\sum^K_{j=1}A_{i_j}) - (i_K - i_1)^2$$
Maciej wyznaczył Ciebie do znalezienia najlepszego podciągu ziemniaków, a w zamian podzieli się z Tobą pysznymi frytkami. Pomóż mu i napisz program, który wyznaczy maksymalną możliwą jakość podciągu K ziemniaków.
Wejście
W pierwszym wierszu standardowego wejścia znajdują się dwie liczby naturalne N, K oznaczające długość grządki ziemniaków oraz liczbę ziemniaków potrzebnych Maciejowi. W następnym wierszu następuje N nieujemnych liczb całkowitych A1, A2, …, AN oznaczających okazałości kolejnych ziemniaków w grządce.
Wyjście
W pierwszym (jedynym) wierszu standardowego wyjścia powinna się znaleźć jedna liczba całkowita oznaczająca maksymalną możliwą jakość podciągu K ziemniaków.
Ograniczenia
2 ≤ K ≤ N ≤ 200 000, 1 ≤ Ai ≤ 1012.
Przykład
Input | Output | Explanation |
|
|
Wybierając drugiego, czwartego i piątego ziemniaka otrzymamy jakość 8 + 3 + 7 − 32 = 9. |