Cod sursa(job #1069215)

Utilizator florin.elfusFlorin Elfus florin.elfus Data 29 decembrie 2013 16:31:28
Problema Euro Scor 30
Compilator cpp Status done
Runda Lista lui wefgef Marime 1.21 kb
#include <stdio.h>
#include <math.h>

long long S[35000], dp[35000];
int st[35000];
const long long INF = 1LL << 55;

struct convexHullTrick {
    long long m, n;

    convexHullTrick() {
        m = 0;
        n = 0;
    };

    convexHullTrick(long long _m, long long _n) {
        m = _m;
        n = _n;
    }

    long long get(int x) {
        return m * x + n;
    }

    long long better(convexHullTrick other) {
        if (m > other.m)
            return INF;
        if (m == other.m)
            return n < other.n ? -INF : INF;
        return ceil((double)(n - other.n) / (other.m - m));
    }
} L[35000];

int main() {
    freopen("euro.in", "r", stdin);
    freopen("euro.out", "w", stdout);

    int N, T;
    scanf("%d%d", &N, &T);
    for (int i = 1; i <= N; ++i) {
        int tmp;
        scanf("%d", &tmp);
        S[i] = S[i - 1] + tmp;
    }

    st[0] = 1;
    for (int i = 1; i <= N; ++i) {
        while (st[0] > 1 && L[st[st[0]]].better(L[st[st[0] - 1]]) <= i)
            --st[0];
        dp[i] = L[st[st[0]]].get(i) + i * S[i] - T;
        L[i] = convexHullTrick(-S[i], dp[i]);
        st[++st[0]] = i;
    }

    printf("%lld", dp[N]);
    return 0;
}