[SDOI2016]征途

题面描述

传送门

思路

方法一

方差

$$s^2=\frac{{\sum }_{i=1}^k(v_i-\bar{v}{)}^2}{k}$$

$$\bar{v}=\frac{{\sum }_{i=1}^k{v}_i}{k}$$

$$s^2=\frac{{\sum }_{i=1}^k(\frac{1}{k}{\sum }_{j=1}^k{v}_j-v_i{)}^2}{k}$$

$$s^2=\frac{{\sum }_{i=1}^k{v}_i^2+k\ast (\frac{{\sum }_{j=1}^k{v}_j}{k}{)}^2-2\ast \frac{({\sum }_{j=1}^k{v}_j{)}^2}{k}}{k}$$

$$s^2=\frac{{\sum }_{i=1}^k{v_i}^2+\frac{({\sum }_{j=1}^k{v}_j{)}^2}{k}-2\ast \frac{({\sum }_{j=1}^k{v}_j{)}^2}{k}}{k}$$

$$s^2{k}^2=k\ast \sum _{i=1}^k{v_i}^2-(\sum _{i=1}^k{v}_i{)}^2$$

状态转移方程

最终答案为$s^2{m}^2=m\ast {\sum }_{i=1}^m{v_i}^2-({\sum }_{i=1}^m{v}_i{)}^2$

$({\sum }_{i=1}^m{v}_i{)}^2$是一个定值。

只需计算$$\sum _{i=1}^m{v_i}^2$$

设$s$为长度前缀和，则

$$F_{i,p+1}=\min (F_{j,p}+(s_i-s_j{)}^2)(i>j)$$

决策单调性

$$F_{j,p}+(s_i-s_j{)}^2\ge F_{k,p}+(s_i-s_k{)}^2$$

对于未来状态$t$，证明:

$$F_{j,p}+(s_t-s_j{)}^2\ge F_{k,p}+(s_t-s_k{)}^2\ast m$$

$s_t=s_i+val(val>0)$

只需证明:

$$-s_j\ge -s_k$$

$s_j<s_k$，证毕。

踢队头

$$F_{j,p}+(s_i-s_j{)}^2\ge F_{k,p}+(s_i-s_k{)}^2$$

$$F_{j,p}-2\ast s_i\ast s_j+{s_j}^2\ge F_{k,p}-2\ast s_i\ast s_k+{s_k}^2$$

$$F_{k,p}-F_{j,p}+{s_k}^2-{s_j}^2\le 2\ast s_i$$

$$calc(j,k)=\frac{F_{k,p}-F_{j,p}+{s_k}^2-{s_j}^2}{s_k-s_j}\le 2\ast s_i$$

$k$优于$j$

即

$$calc(q_{head},q_{head+1})\le 2\ast s_i$$

$q_{head+1}$优于$q_{head}$，踢去$head$

维护一个$calc(q_{head},q_{head+1})\ge 2\ast s_i$队列

同时我们可以发现，$calc(q_{head},q_{head+1})$随$2\ast s_i$的增大，$head$不断增大，而不断增大。

因此，队列维护的应该是一个下凸壳。

踢队尾

由于维护一个下凸壳，

所以

$$calc(q_{tail-1},q_{tail})\le calc(q_{tail},i-m)$$

反之，

当

$$calc(q_{tail-1},q_{tail})\ge calc(q_{tail},i-m)$$

踢去队尾。

方法一 code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<iostream>
#define gc getchar()
#define ll long long 
using namespace std;
const int N=3e3+10;
template<class orz>
inline void qr(orz &x)
{
    x=0;int f=1;char c=gc;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=gc;}
    while(c>='0'&&c<='9'){x=x*10+(c^48);c=gc;}
    x*=f;
}
void qw(ll x)
{
    if(x<0)x=-x,putchar('-');
    if(x/10)qw(x/10);
    putchar(x%10+48);
}
ll f[N],g[N],ave;int s[N];
inline double calc(int j,int k)
{
    return (double)(g[k]-g[j]+s[k]*s[k]-s[j]*s[j])/(s[k]-s[j]);
}
int q[N],l,r;
int main()
{
    int n,m;qr(n);qr(m);
    for(int i=1;i<=n;i++)qr(s[i]),s[i]+=s[i-1];
    for(int i=1;i<=n;i++)f[i]=s[i]*s[i];
    for(int j=2;j<=m;j++)
    {
        memcpy(g,f,n<<3);l=1;r=1;q[1]=j-1;
        for(int i=j;i<=n;i++)
        {
            while(l<r&&calc(q[l],q[l+1])<=2.0*s[i])++l;
            while(l<r&&calc(q[r],i)<=calc(q[r-1],q[r]))--r;
            q[++r]=i;f[i]=g[q[l]]+(s[i]-s[q[l]])*(s[i]-s[q[l]]);
        }
    }
    qw(f[n]*m-s[n]*s[n]);
    return 0;
}

方法二

设$s$为长度前缀和

仔细观察发现，貌似一天所走的路(从$j+1$走到$i$)对最终答案的贡献为:

$$(s_i-s_j-ave{)}^2\ast m$$

那么，方程就为:

$$F_{i,p}=\min (F_{j.p-1}+(s_i-s_j-ave{)}^2\ast m)$$

决策单调性

$$F_{j,p}+(s_i-s_j-ave{)}^2\ast m\ge F_{k,p}+(s_i-s_k-ave{)}^2\ast m$$

对于未来状态$t$，证明:

$$F_{j,p}+(s_t-s_j-ave{)}^2\ge F_{k,p}+(s_t-s_k-ave{)}^2\ast m$$

$s_t=s_i+val(val>0)$

只需证明:

$$-s_j\ge -s_k$$

$s_j<s_k$，证毕。

踢队头

$$F_{j,p}+(s_i-s_j-ave{)}^2\ast m\ge F_{k,p}+(s_i-s_k-ave{)}^2\ast m$$

$$F_{j,p}+m\ast ((s_i-s_j{)}^2-2\ast (s_i-s_j)\ast ave+av{e}^2)\ge F_{k,p}+m\ast ((s_i-s_k{)}^2-2\ast (s_i-s_k)\ast ave+av{e}^2)$$

$$F_{j,p}+m\ast (-2\ast s_i\ast s_j+{s_j}^2+2\ast s_j\ast ave)\ge F_{k,p}+m\ast (-2\ast s_i\ast s_k+{s_k}^2+2\ast s_k\ast ave))$$

$$F_{k,p}-F_{j,p}+m\ast ({s_k}^2-{s_j}^2+2\ast ave\ast (s_k-s_j))\le 2\ast m\ast s_i\ast (s_k-s_j)$$

$$calc(j,k)=\frac{F_{k,p}-F_{j,p}+m\ast ({s_k}^2-{s_j}^2+2\ast ave\ast (s_k-s_j))}{s_k-s_j}\le 2\ast m\ast s_i$$

$k$优于$j$

即

$$calc(q_{head},q_{head+1})\le 2\ast m\ast s_i$$

$q_{head+1}$优于$q_{head}$，踢去$head$

维护一个$calc(q_{head},q_{head+1})\ge 2\ast m\ast s_i$队列

同时我们可以发现，$calc(q_{head},q_{head+1})$随$2\ast m\ast s_i$的增大，$head$不断增大，而不断增大。

因此，队列维护的应该是一个下凸壳。

踢队尾

由于维护一个下凸壳，

所以

$$calc(q_{tail-1},q_{tail})\le calc(q_{tail},i-m)$$

反之，

当

$$calc(q_{tail-1},q_{tail})\ge calc(q_{tail},i-m)$$

踢去队尾。

方法2 code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<iostream>
#define gc getchar()
#define ll long long 
using namespace std;
const int N=3e3+10;
template<class orz>
inline void qr(orz &x)
{
    x=0;int f=1;char c=gc;
    while(c<'0'||c>'9'){if(c=='-')f=-1;c=gc;}
    while(c>='0'&&c<='9'){x=x*10+(c^48);c=gc;}
    x*=f;
}
void qw(ll x)
{
    if(x<0)x=-x,putchar('-');
    if(x/10)qw(x/10);
    putchar(x%10+48);
}
double f[N],g[N],ave;int s[N];
inline double calc(int j,int k)
{
    return (double)(g[k]-g[j]+s[k]*s[k]-s[j]*s[j]+2*ave*(s[k]-s[j]))/(s[k]-s[j]);
}
int q[N],l,r;
int main()
{
    int n,m;qr(n);qr(m);
    for(int i=1;i<=n;i++)qr(s[i]),s[i]+=s[i-1];ave=(double)s[n]/m;
    for(int i=1;i<=n;i++)f[i]=(s[i]-ave)*(s[i]-ave);
    for(int j=2;j<=m;j++)
    {
        memcpy(g,f,sizeof(g));l=1;r=1;q[1]=j-1;
        for(int i=j;i<=n;i++)
        {
            while(l<r&&calc(q[l],q[l+1])<=2.0*s[i])++l;
            while(l<r&&calc(q[r],i)<=calc(q[r-1],q[r]))--r;
            q[++r]=i;f[i]=g[q[l]]+(s[i]-s[q[l]]-ave)*(s[i]-s[q[l]]-ave);
        }
    }
    printf("%.0lf\n",f[n]*m);
    return 0;
}