【题解】洛谷P4072 [SDOI2016] 征途 二维斜率DP

给出n,m(3000)，和一个长为n的数列(和<30000)，将其分成连续的m段，使得每段和的方差最小。
输出$方差\ast m^2$（一定是整数）。

---

利用方差公式，$答案=m\Sigma x_i^2-(\Sigma x{)}^2$，其中$x_i$表示第$i$段的数字之和。

所以目标是让$\Sigma x_i^2$最小。

设$d_{ij}$表示把前$i$个数字分成$j$段时的最小代价，$d_{00}=0$.
d i j = m i n { d k , j − 1 + ( s i − s k ) 2 } d_{ij}=min\{d_{k,j-1}+(s_i-s_k)^2\} dij​=min{dk,j−1​+(si​−sk​)2}，$j-1<=k<i$，$s_i$表示前$i$个数的和。

先按$j$外层循环，内层再按$i$循环。

---

整理得， d i = s i 2 + m i n { d k + s k 2 − 2 s i s k } d_i=s_i^2+min\{d_k+s_k^2-2s_is_k\} di​=si2​+min{dk​+sk2​−2si​sk​}

设$a<b<i$且从$b$转移优于从$a$转移，那么：
$$\frac{(d_b+s_b^2)-(d_a+s_a^2)}{s_b-s_a}<2s_i$$

由斜率优化原理，$y_i=d_i+s_i^2，x_i=s_i，k=2s_i$，维护下凸包，斜率递增。

---

WA点：

1. subx和suby写混
2. 分析过程中出错，目标斜率少了常数2

---

/* LittleFall : Hello! */
#include <bits/stdc++.h>
using namespace std; using ll = long long; inline int read();
const int M = 3016, MOD = 1000000007;

ll sum[M];
ll dp[M][M];
int segs; //做指示变量，因为调用太麻烦
inline ll subx(ll a, ll b)
{
	return sum[b] - sum[a];
}
inline ll suby(ll a, ll b)
{
	return (dp[b][segs-1]+sum[b]*sum[b]) - (dp[a][segs-1]+sum[a]*sum[a]);
}
inline ll cal(int i, int k)
{
	return dp[k][segs-1]+(sum[i]-sum[k])*(sum[i]-sum[k]);
}
int q[M], l, r;
int main(void)
{
	#ifdef _LITTLEFALL_
	freopen("in.txt","r",stdin);
    #endif

	int n = read(), m = read();
	for(int i=1; i<=n; ++i)
		sum[i] = sum[i-1] + read();

	for(int i=1; i<=n; ++i)
		dp[i][1] = sum[i] * sum[i];

	for(segs=2; segs<=m; ++segs)
	{
		l = r = 0; q[r++] = segs-1;
		for(int i=segs; i<=n; ++i)
		{
			while(r-l>=2 && suby(q[l], q[l+1]) <= subx(q[l],q[l+1])*2*sum[i] ) ++l;
			dp[i][segs] = cal(i, q[l]);
			while(r-l>=2 && 
				suby(q[r-2], q[r-1]) * subx(q[r-1],i) >= subx(q[r-2], q[r-1]) * suby(q[r-1], i)
			) --r;
			q[r++] = i;
		}
	}
	cout << m*dp[n][m]-sum[n]*sum[n] << "\n";

    return 0;
}


inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9') {if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}