【HNOI 2008】玩具装箱

传送门

---

Problem

给出一个长度为 $n$ 的序列 { a n } \{a_n\} {an​} 以及常数 $m$，现要将它分成若干段，对于一个 $[i,j]$ 的区间的贡献是 $(j-i+\sum _{k=i}^j{a}_k-m{)}^2$，求总贡献的最小值。

数据范围：$1\le n\le 50000$，$1\le n,a_i\le 10^7$。

---

Solution

我们定义 $S_i$ 为 $a_i$ 的前缀和，$f_i$ 为在 $i$ 到 $i+1$ 分割一次的最小费用。

那么有

f i = min ⁡ j = 1 i − 1 { f j + ( i − j − 1 + ( S i − S j ) − m ) 2 } f_i=\min_{j=1}^{i-1}\{f_j+(i-j-1+(S_i-S_j)-m)^2\} fi​=j=1mini−1​{fj​+(i−j−1+(Si​−Sj​)−m)2}

那我们令 $G_i=S_i+i$（为了方便化简），得：

f i = min ⁡ j = 1 i − 1 { f j + ( G i − G j − ( m + 1 ) ) 2 } f_i=\min_{j=1}^{i-1}\{f_j+(G_i-G_j-(m+1))^2\} fi​=j=1mini−1​{fj​+(Gi​−Gj​−(m+1))2}

于是选择任意的 $k<j<i$，当 $j$ 比 $k$ 优时，有（接下来就是把平方拆开然后化简，就不写了）：

$$\frac{(f_j+G_j^2+2(m+1)G_j)-(f_k+G_k^2+2(m+1)G_k)}{G_j-G_k}<2G_i$$

然后维护一个下凸壳转移即可。

---

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 50005
#define ll long long
using namespace std;
int n,m,Q[N];
ll S[N],G[N],f[N];
ll Squ(ll x)  {return x*x;}
ll ordi(int x)  {return f[x]+Squ(G[x])+2ll*(m+1)*G[x];}
double slope(int x,int y)  {return 1.0*(ordi(y)-ordi(x))/(G[y]-G[x]);}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1,x;i<=n;++i){
		scanf("%d",&x),S[i]=S[i-1]+x,G[i]=S[i]+i;
	}
	int l=0,r=0;
	for(int i=1;i<=n;++i){
		while(l<r&&slope(Q[l],Q[l+1])<=2*G[i])  l++;
		f[i]=f[Q[l]]+Squ(G[i]-G[Q[l]]-(m+1));
		while(l<r&&slope(Q[r-1],Q[r])>=slope(Q[r],i))  r--;
		Q[++r]=i;
	}
	printf("%lld\n",f[n]);
	return 0;
}