Homework Introduction
//tail=0 head=0 while(tail>head)
//tail=-1 head=0 while(tail>=head)
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N = 5e4 + 5;
int n, L, sum[N], q[N], tail, head;
int f[N];
int A(int i) {
return sum[i] + i;
}
int B(int i) {
return sum[i] + i + L + 1;
}
int X(int i) {
return B(i);
}
int Y(int i) {
return f[i] + B(i) * B(i);
}
double k(int i, int j) {
return double(Y(j) - Y(i)) / double(X(j) - X(i));
}
signed main() {
cin >> n >> L;
for (int i = 1; i <= n; i++) {
cin >> sum[i];
sum[i] += sum[i - 1];
}
for (int i = 1; i <= n; i++) {
while (tail > head && 2 * A(i) > k(q[head], q[head + 1]))
head++;
int j = q[head];
f[i] = f[j] + (A(i) - B(j)) * (A(i) - B(j));
while (tail > head && k(i, q[tail - 1]) < k(q[tail], q[tail - 1]))
--tail;
q[++tail] = i;
}
cout << f[n] << endl;
return 0;
}
- Status
- Done
- Problem
- 12
- Open Since
- 2025-7-14 0:00
- Deadline
- 2025-8-7 23:59
- Extension
- 24 hour(s)