[洛谷P4072] SDOI2016 征途

问题描述

Pine开始了从S地到T地的征途。

从S地到T地的路可以划分成n段，相邻两段路的分界点设有休息站。

Pine计划用m天到达T地。除第m天外，每一天晚上Pine都必须在休息站过夜。所以，一段路必须在同一天中走完。

Pine希望每一天走的路长度尽可能相近，所以他希望每一天走的路的长度的方差尽可能小。

帮助Pine求出最小方差是多少。

设方差是v，可以证明，(v imes m^2)是一个整数。为了避免精度误差，输出结果时输出(v imes m^2)。

输入格式

第一行两个数 n、m。

第二行 n 个数，表示 n 段路的长度

输出格式

一个数，最小方差乘以 (m^2) 后的值 。

样例输入

5 2
1 2 5 8 6

样例输出

36

说明

对于 (30\%) 的数据，(1 le n le 10)。

对于 (60\%) 的数据，(1 le n le 100)。

对于 (100\%) 的数据，(1 le n le 3000)。

保证从 S 到 T 的总路程不超过 30000 。

解析

首先，我们需要化简方差的式子，

[egin{align}s^2 &=frac{sum_{i=1}^{m}(overline v-v_i)^2}{m}\ &=frac{moverline v^2-2overline v(v_1+v_2+...+v_m)+(v_1^2+v_2^2+...+v_m^2)}{m}\ &=frac{mfrac{(sum_{i=1}^{m}v_i)^2}{m^2}-2frac{(sum_{i=1}^{m}v_i)^2}{m}+v_1^2+v_2^2+...+v_m^2}{m}\end{align} ]

所以

[s^2 imes m^2=-(sum_{i=1}^{m}v_i)^2+m(v_1^2+...+v_m^2) ]

所以，我们需要把路程划分为m个部分，使(v_1^2+...+v_m^2)最小。这个可以用动态规划来完成。设(f[i][j])表示将前i个数划分成j段的最小值。我们有如下状态转移方程：

[f[i][j]=max(f[k][j-1]+(sum[i]-sum[k])^2) ]

然后这个转移方程可以用斜率优化。

代码

#include <iostream>
#include <cstdio>
#define int long long
#define N 3002
using namespace std;
int n,m,i,j,v[N],sum[N],f[N][N],q[N],head,tail;
int read()
{
	char c=getchar();
	int w=0;
	while(c<'0'||c>'9') c=getchar();
	while(c<='9'&&c>='0'){
		w=w*10+c-'0';
		c=getchar();
	}
	return w;
}
double k(int x,int i,int j)
{
	return 1.0*((f[i][x]+sum[i]*sum[i])-(f[j][x]+sum[j]*sum[j]))/(sum[i]-sum[j]);
}
signed main()
{
	n=read();m=read();
	for(i=1;i<=n;i++){
		v[i]=read();
		sum[i]=sum[i-1]+v[i];
		f[i][1]=sum[i]*sum[i];
	}
	for(j=2;j<=m;j++){
		head=tail=1;
		q[1]=j-1;
		for(i=j;i<=n;i++){
			while(head<tail&&k(j-1,q[head],q[head+1])<2*sum[i]) head++;
			int x=q[head];
			f[i][j]=f[x][j-1]+(sum[i]-sum[x])*(sum[i]-sum[x]);
			while(head<tail&&k(j-1,q[tail],i)<k(j-1,q[tail],q[tail-1])) tail--;
			q[++tail]=i;
		}
	}
	printf("%lld
",m*f[n][m]-sum[n]*sum[n]);
	return 0;
}