题解
有人管它叫带权二分,有人管它叫dp凸优化,有人管它叫wqs二分……
延伸出来还有zgl分治,xjp¥!%#!@#¥!#
当我没说
我们拆个式子,很容易发现所求的就是
\(m\sum_{i = 1}^{m}s_{i}^2 - sum^{2}\)
然后去掉常数我们只要求\(\sum_{i = 1}^{m}s_{i}^2\)的最小值
然而,我们需要m个?
我们发现,这个东西随着选的个数增多,越来越少,并且少得越来越慢(斜率变大,斜率是负的!)
我们二分最后一次的斜率,选一次减少q,最后能取到的最小值用了几天,如果恰好等于m那么就是我们需要的,这个dp显然就是用斜率优化一下就OK了
当然,我们很可能出现,m = 4,前一个二分到最小值天数为3,再 + 1就变成最小值天数为5了= =,事实上二分的边界再多一个左右斜率相等就行了
代码
#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <map>
//#define ivorysi
#define pb push_back
#define space putchar(' ')
#define enter putchar('\n')
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define mo 974711
#define MAXN 3005
#define RG register
using namespace std;
typedef long long int64;
typedef double db;
template<class T>
void read(T &res) {
res = 0;char c = getchar();T f = 1;
while(c < '0' || c > '9') {
if(c == '-') f = -1;
c = getchar();
}
while(c >= '0' && c <= '9') {
res = res * 10 + c - '0';
c = getchar();
}
res *= f;
}
template<class T>
void out(T x) {
if(x < 0) {putchar('-');x = -x;}
if(x >= 10) {
out(x / 10);
}
putchar('0' + x % 10);
}
int N,M,que[MAXN],ql,qr,cnt[MAXN];
int64 S[MAXN],dp[MAXN],V,X[MAXN],Y[MAXN];
int calc(int a,int b) {
return dp[a] + (S[b] - S[a]) * (S[b] - S[a]) - V;
}
bool slope(int a,int b,int c) {
return (X[c] - X[a]) * (Y[b] - Y[a]) - (X[b] - X[a]) * (Y[c] - Y[a]) >= 0;
}
int Check(int64 mid) {
V = mid;
ql = 1,qr = 1;
cnt[0] = 0;dp[0] = 0;
que[1] = 0;
for(int i = 1 ; i <= N ; ++i) {
while(ql < qr) {
if(calc(que[ql],i) > calc(que[ql + 1],i)) ++ql;
else break;
}
dp[i] = calc(que[ql],i);cnt[i] = cnt[que[ql]] + 1;
X[i] = S[i];Y[i] = dp[i] + S[i] * S[i];
while(ql < qr) {
if(slope(que[qr - 1],que[qr],i)) --qr;
else break;
}
que[++qr] = i;
}
return cnt[N];
}
void Solve() {
read(N);read(M);
for(int i = 1 ; i <= N ; ++i) read(S[i]);
for(int i = 1 ; i <= N ; ++i) S[i] += S[i - 1];
int64 L = -1000000000,R = 1000000000;
while(L <= R) {
int64 MID = (L + R + 1) >> 1;
int x = Check(MID);
if(x == M || L == R) {
int64 ans = M * (dp[N] + MID * M) - S[N] * S[N];
out(ans);enter;return;
}
if(x > M) R = MID - 1;
else L = MID;
}
}
int main() {
#ifdef ivorysi
freopen("f1.in","r",stdin);
#endif
Solve();
return 0;
}