【BZOJ4518】【SDOI2016】—征途（斜率优化dp）

传送门

题意：有$n$个数，要把这些数分成连续的$m$段使方差最小

由于$s^2=\frac{{\sum }_{i=1}^n(d-d_i{)}^2}{m}$

则$s^2\ast m^2=m\ast {\sum }_{i=1}^n(d-d_i{)}^2$

令$su{m}_k={\sum }_{i=1}^k{d}_i$

因为$d=\frac{{\sum }_{i=1}^n{d}_i}{m}=\frac{su{m}_n}{m}$
化简得${\sum }_{i=1}^n(d_i^2)-({\sum }_{i=1}^n{d}_i{)}^2$

也就是要最小化${\sum }_{i=1}^n(d_i^2)$

那就有一个很显然的$O(n^{2}m)$的$dp$
$f[j][i]$表示前$i$个分$j$段的最小值

$f[k][i]=Min(f[k-1][j]+(su{m}_i-su{m}_j{)}^2)$

考虑优化
如果一个决策点$a$比$b$优
则
$f[k][a]+(su{m}_i-su{m}_a{)}^2<f[k][b]+(su{m}_i-su{m}_b{)}^2$

化简就变成了$$\frac{f_a+su{m}_a^2-f_b-su{m}_b^2}{su{m}_a-su{m}_b}\le 2\ast su{m}_i$$

把$f_a+su{m}_a^2$看做$x$,$su{m}_a$看做$y$
因为$sum$单调
也就是说可以维护一个斜率递增的序列

答案也就从当前最小的斜率转移过来就行了

#include<bits/stdc++.h>
using namespace std;
#define int long long
inline int read(){
	char ch=getchar();
	int res=0,f=1;
	while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
	while(isdigit(ch))res=(res+(res<<2)<<1)+(ch^48),ch=getchar();
	return res*f;
}
const int N=3005;
const double eps=1e-9;
int sum[N],f[N][N],d[N],n,m;
double q[N];
inline int P(int x){
	return x*x;
}
inline double slope(int p,int a,int b){
	return (double)(1.0*f[p][a]+P(sum[a])-f[p][b]-P(sum[b]))/(sum[a]-sum[b]);
}
signed main(){
	n=read(),m=read();
	for(int i=1;i<=n;i++)d[i]=read(),sum[i]=sum[i-1]+d[i],f[1][i]=P(sum[i]);
	for(int j=2;j<=m;j++){
		int hd=1,tl=0;
		for(int i=1;i<=n;i++){
			while(hd<tl&&slope(j-1,q[hd],q[hd+1])<2*sum[i])hd++;
			int now=q[hd];
			f[j][i]=f[j-1][now]+P(sum[i]-sum[now]);
			while(hd<tl&&slope(j-1,q[tl-1],q[tl])>slope(j-1,q[tl-1],i))tl--;
			q[++tl]=i;
		}
	}
	int ans=m*f[m][n]-P(sum[n]);
	cout<<ans;
}