luogu P4072 [SDOI2016]征途

背景：

$hehe...$

题目传送门：

https://www.luogu.org/problemnew/show/P4072

题意：

$n$个数$a_i$，分成$m$段，使方差（$v$）尽可能小，求$v\ast m^2$。

思路：

推公式吧。
假设每一段的和为$x_1,x_2,...,x_m$。
设$su{m}_k={\sum }_{i=1}^k{a}_i$。
另$x=\frac{{\sum }_{i=1}^m{x}_i}{m}$则有：
$v=\frac{{\sum }_{i=1}^m(x_i-x{)}^2}{m}$。
所以有：$v=\frac{{\sum }_{i=1}^m(x_i-x{)}^2}{m}$

我们可以知道$x=\frac{{\sum }_{i=1}^m{x}_i}{m}=\frac{{\sum }_{i=1}^n{a}_i}{m}$，所以有：

$\frac{{\sum }_{i=1}^m(x_i-x{)}^2}{m}$
$=\frac{{\sum }_{i=1}^m(x_i-\frac{{\sum }_{i=1}^n{a}_i}{m}{)}^2}{m}$
$=\frac{{\sum }_{i=1}^m(x_i-\frac{su{m}_n}{m}{)}^2}{m}$
$=\frac{{\sum }_{i=1}^m(x_i^2-2x_i\frac{su{m}_n}{m}+\frac{su{m}_n^2}{m^2})}{m}$
$=\frac{{\sum }_{i=1}^m(x_i^2)-{\sum }_{i=1}^m(2x_i\frac{su{m}_n}{m})+{\sum }_{i=1}^m(\frac{su{m}_n^2}{m^2}))}{m}$
$=\frac{{\sum }_{i=1}^m{x}_i^2}{m}-\frac{{\sum }_{i=1}^{m}2x_i\frac{su{m}_n}{m}}{m}+\frac{{\sum }_{i=1}^m\frac{su{m}_n^2}{m^2}}{m}$
$=\frac{{\sum }_{i=1}^m{x}_i^2}{m}-\frac{2su{m}_n{\sum }_{i=1}^m{x}_i}{m^2}+\frac{{\sum }_{i=1}^{m}su{m}_n^2}{m^3}$
$=\frac{{\sum }_{i=1}^m{x}_i^2}{m}-\frac{2su{m}_n^2}{m^2}+\frac{m\ast su{m}_n^2}{m^3}$
$=\frac{{\sum }_{i=1}^m{x}_i^2}{m}-\frac{2su{m}_n^2}{m^2}+\frac{su{m}_n^2}{m^2}$
$=\frac{{\sum }_{i=1}^m{x}_i^2}{m}-\frac{su{m}_n^2}{m^2}$
由于只有${\sum }_{i=1}^m{x}_i^2$是不定值，所以让其最小即可。

容易想到常规做法：设$f_{i,l}$表示前$i$个数用$l$次来选的最小值。
显然$f_{i,l}={\min }_{j=1}^{i-1}{f}_{j,l-1}+(su{m}_i-su{m}_j{)}^2$。
则斜率方程为：$\frac{f_{j,l-1}-f_{k,l-1}+su{m}_j^2-su{m}_k^2}{su{m}_j-su{m}_k}<2su{m}_i$

最后套回去即可。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
	int n,m;
	LL a[3010],sum[3010],f[3010][3010];
	int que[3010];
LL calc1(int x,int y)
{
	return sum[y]-sum[x];
}
LL calc2(int x,int y,int z)
{
	return (f[y][z-1]+sum[y]*sum[y])-(f[x][z-1]+sum[x]*sum[x]);
}
double calc(int x,int y,int z)
{
	return (double)(calc2(x,y,z))/(double)(calc1(x,y));
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		sum[i]=sum[i-1]+a[i];
	}
	for(int i=1;i<=n;i++)
		f[i][1]=sum[i]*sum[i];
	for(int l=2;l<=m;l++)
	{
		int head=1,tail=1;
		que[1]=0;
		for(int i=1;i<=n;i++)
		{
			while(head<tail&&calc(que[head],que[head+1],l)<=(double)2*sum[i]) head++;
			f[i][l]=f[que[head]][l-1]+calc1(que[head],i)*calc1(que[head],i);
			while(head<tail&&calc(que[tail],i,l)<=calc(que[tail-1],que[tail],l)) tail--;
			que[++tail]=i;
		}
	}
	printf("%lld",f[n][m]*m-sum[n]*sum[n]);
}