[洛谷P4072] SDOI2016 征途

问题描述

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

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

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

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

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

设方差是v，可以证明，\(v\times m^2\)是一个整数。为了避免精度误差，输出结果时输出\(v\times 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 。

解析

首先，我们需要化简方差的式子，
\[ \begin{align}s^2 &=\frac{\sum_{i=1}^{m}(\overline v-v_i)^2}{m}\\ &=\frac{m\overline v^2-2\overline v(v_1+v_2+...+v_m)+(v_1^2+v_2^2+...+v_m^2)}{m}\\ &=\frac{m\frac{(\sum_{i=1}^{m}v_i)^2}{m^2}-2\frac{(\sum_{i=1}^{m}v_i)^2}{m}+v_1^2+v_2^2+...+v_m^2}{m}\\\end{align} \]
所以
\[ s^2\times 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 
#include 
#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