Description
Pine开始了从S地到T地的征途。
从S地到T地的路可以划分成n段,相邻两段路的分界点设有休息站。
Pine计划用m天到达T地。除第m天外,每一天晚上Pine都必须在休息站过夜。所以,一段路必须在同一天中走完。
Pine希望每一天走的路长度尽可能相近,所以他希望每一天走的路的长度的方差尽可能小。
帮助Pine求出最小方差是多少。
设方差是v,可以证明,v×m^2是一个整数。为了避免精度误差,输出结果时输出v×m^2。
Solution
我是来复习斜率优化板子的。。。
这题其实是要求
∑x2i−Sum2
∑
x
i
2
−
S
u
m
2
的最小值。
然后列出转移方程后,颓一下式子,斜率优化就行了。。
Code
/************************************************
* Au: Hany01
* Date: Apr 9th, 2018
* Prob: [BZOJ4518][SDOI2016] 征途
* Email: hany01@foxmail.com
************************************************/
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int, int> PII;
#define File(a) freopen(a".in", "r", stdin), freopen(a".out", "w", stdout)
#define rep(i, j) for (register int i = 0, i##_end_ = (j); i < i##_end_; ++ i)
#define For(i, j, k) for (register int i = (j), i##_end_ = (k); i <= i##_end_; ++ i)
#define Fordown(i, j, k) for (register int i = (j), i##_end_ = (k); i >= i##_end_; -- i)
#define Set(a, b) memset(a, b, sizeof(a))
#define Cpy(a, b) memcpy(a, b, sizeof(a))
#define x first
#define y second
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define ALL(a) (a).begin(), (a).end()
#define SZ(a) ((int)(a).size())
#define INF (0x3f3f3f3f)
#define INF1 (2139062143)
#define Mod (1000000007)
#define debug(...) fprintf(stderr, __VA_ARGS__)
#define y1 wozenmezhemecaia
template <typename T> inline bool chkmax(T &a, T b) { return a < b ? a = b, 1 : 0; }
template <typename T> inline bool chkmin(T &a, T b) { return b < a ? a = b, 1 : 0; }
inline int read()
{
register int _, __; register char c_;
for (_ = 0, __ = 1, c_ = getchar(); c_ < '0' || c_ > '9'; c_ = getchar()) if (c_ == '-') __ = -1;
for ( ; c_ >= '0' && c_ <= '9'; c_ = getchar()) _ = (_ << 1) + (_ << 3) + (c_ ^ 48);
return _ * __;
}
const int maxn = 3005;
int n, m, sum[maxn], a[maxn];
LL f[maxn], g[maxn];
inline double Slope(int j, int k) {
return ((g[j] + sum[j] * sum[j]) - (g[k] + sum[k] * sum[k])) * 1. / (sum[j] - sum[k]);
}
inline void DP(int cur)
{
static int q[maxn], head, tail;
head = 1, tail = 0;
For(i, cur, n) {
q[++ tail] = i - 1;
while (head + 1 < tail && Slope(q[tail - 2], q[tail - 1]) >= Slope(q[tail - 1], q[tail])) q[tail - 1] = q[tail], -- tail;
while (head < tail && Slope(q[head], q[head + 1]) <= 2 * sum[i]) ++ head;
register int j = q[head];
f[i] = g[j] + (sum[i] - sum[j]) * (sum[i] - sum[j]);
}
}
int main()
{
#ifdef hany01
File("bzoj4518");
#endif
n = read(), m = read();
For(i, 1, n) sum[i] = sum[i - 1] + (a[i] = read());
For(i, 1, n) g[i] = INF;
For(i, 1, m) DP(i), swap(f, g);
printf("%lld\n", g[n] * (LL)m - sum[n] * sum[n]);
return 0;
}
//Error(s):
//最后输出的是g而不是f
//g没有赋最大值
//秋风起兮白云飞,草木黄落兮雁南归。
// -- 刘彻《秋风辞》