Cod sursa(job #2803079)

Utilizator Mihai145Oprea Mihai Adrian Mihai145 Data 19 noiembrie 2021 14:00:10
Problema Euro Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.65 kb
//
// Created by Mihai145 on 11/19/2021.
//

#include <fstream>

using namespace std;

ifstream cin("euro.in");
ofstream cout("euro.out");

const long long NMAX = 35000;

long long N, T;
long long sp[NMAX + 2], dp[NMAX + 2];

struct Line {
    long long m, n;
    long long eval(long long x) const {
        return m * x + n;
    }
};

Line segm_tree[4 * NMAX];

void insert_line(long long node, long long l, long long r, Line ins) {
    if(l == r) {
        return;
    }

    long long mid = (l + r) >> 1;

    bool better_left = ins.eval(l) > segm_tree[node].eval(l),
        better_mid = ins.eval(mid) > segm_tree[node].eval(mid);

    if(better_mid) {
        swap(segm_tree[node], ins);
    }

    if(better_left != better_mid) {
        insert_line(node << 1, l, mid, ins);
    } else {
        insert_line((node << 1) + 1, mid + 1, r, ins);
    }
}

long long query(long long node, long long l, long long r, long long x) {
    if(l == r) {
        return segm_tree[node].eval(x);
    }

    long long mid = (l + r) >> 1;
    long long candidate = 0;

    if(x <= mid) {
        candidate = query(node << 1, l, mid, x);
    } else {
        candidate = query((node << 1) + 1, mid + 1, r, x);
    }

    return max(segm_tree[node].eval(x), candidate);
}

int main() {
    cin >> N >> T;

    for(long long i = 1; i <= N; i++) {
        cin >> sp[i];
        sp[i] += sp[i - 1];
    }

    insert_line(1, 1, N, {0, 0});

    for(long long i = 1; i <= N; i++) {
        dp[i] = sp[i] * i - T + query(1, 1, N, i);
        insert_line(1, 1, N, {-sp[i], dp[i]});
    }

    cout << dp[N] << '\n';
    return 0;
}