Посчитать неприступность крепости

@ggbetgamer22 · Регистрация: 09.03.2021

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

помогите пожалуйста дописать код, я только смог написать код который выводит максимальную защищенность крепости, а как сделать с расположением башен не могу понять
Башня
Петя в очередной раз купил себе набор из кубиков. На этот раз он выстроил из них настоящую крепость — последовательность из N столбиков, высота каждого столбика составляет Ai кубиков.

Вскоре ему стало интересно, насколько его крепость защищена от жуликов и воров. Для этого он ввел понятие башни. Башней называется любая последовательность из K столбиков подряд (где K — любимое число Пети). Защищённость башни определяется как суммарная высота всех столбиков этой башни (чем она больше, тем громаднее и ужаснее она кажется), умноженная на минимум высоты столбиков башни (т.к. враги, очевидно, будут пытаться проникнуть через самое слабое место башни). Неприступность крепости определяется как сумма защищённостей каждой из башен.

Петя решил как можно скорее посчитать, какова же неприступность его крепости. Однако вскоре он понял, что недостаточно знать высоту каждого из столбиков. В зависимости от того, как сгруппировать столбики в башни, получится разный результат. В различных вариантах группировки часть столбиков могут не принадлежать ни одной из башен. Разумеется, Петя выберет то разбиение на башни, при котором неприступность будет максимальна.

Петя успешно справился со своей задачей, но теперь Правительство Флатландии решило защитить свой горный курорт. Правительство уже построило крепость из кубиков (просто кубики были побольше). Теперь вы должны помочь Правительству посчитать неприступность этой крепости. Единственная трудность состоит в том, что у Правительства было очень много денег, и поэтому крепость была построена очень длинная.

Входные данные

В первой строке содержатся число N — количество столбиков в крепости и число K — любимое число Пети (1 ≤ K ≤ N ≤ 100 000). Далее в следующей строке содержатся N целых чисел, обозначающих Ai (1 ≤ Ai ≤ 106).

Выходные данные

В первой строке выведите число Q — количество башен в оптимальном разбиении. Далее выведите Q чисел — номера первых столбиков каждой башни.

Примеры
Ввод
8 3
1 2 3 4 1 6 7 8
Вывод
2
2 6
Ввод
1 1
1
Вывод
1
1
Ввод
2 1
1 1000000
Вывод
2
1 2

C++

#include <iostream>
#include <deque>
#include <vector>
using namespace std;
int main() {
    int n, k;
    cin >> n >> k;
    vector <int> pref(n + 1, 0), a(n), res, p, prefk(n + 1, 0);
    for (int i = 0; i < n; i++) {
        cin >> a[i];
        pref[i + 1] = pref[i] + a[i];
    }
    prefk = pref;
    deque <int> deque;
    for (int i = 0; i < n; ++i) {
        if (i >= k && deque.front() == i - k)
            deque.pop_front();
        while (deque.size() && a[deque.back()] >= a[i])
            deque.pop_back();
        deque.push_back(i);
        if (i >= k - 1) {
            res.push_back(deque.front());
            prefk[i + 1] -= pref[i + 1 - k];
            prefk[i + 1] *= a[res.back()]; 
        }
    }
    vector <int> ans;
    for (int i = k; i <= n; i++) {
        if (i - k < k - 1)
            prefk[i] = max(prefk[i - 1], prefk[i]);
        else 
            prefk[i] = max(prefk[i - 1], prefk[i - k] + prefk[i]);
    }
    cout << prefk.back();
}

@zss · 19.08.2021, 14:00

Посчитать неприступность крепости

@zss Модератор 13507 / 10757 / 6412 Регистрация: 18.12.2011 Сообщений: 28,712
	19.08.2021, 14:00	2
	Посчитать неприступность крепости 0