标签：src cpp ace 证明 分享 while 开始 帮助 维护
原文地址：http://www.cnblogs.com/GXZlegend/p/6812435.html
题目描述
输入
输出
一个数，最小方差乘以 m^2 后的值
样例输入
5 2
1 2 5 8 6
样例输出
36
题解
斜率优化（不需要二维）
所以只要维护∑m*x[i^2-2*sum*x[i]的最小值即可。
设f[i][j]为前i条路分为j段的∑最小值，那么显然有f[i][j]=f[k][j-1]+m*(sum[i]-sum[k])*(sum[i]-sum[k])-2*sum[n]*(sum[i]-sum[k])。
这样dp时间复杂度为O(n^2*m)，会TLE，需要优化。
将上述dp方程平方展开并移项，得到f[k][j-1]+m*sum[j]^2+2*sum[n]*sum[j]=2*m*sum[i]*sum[j]+f[i][j]-m*sum[i]^2+2*sum[n]*sum[i]
这样可以用斜率优化来优化。
由于第二维j的存在，需要先循环第二维j，再循环第一维i，并将每次的f[i][j-1]与队列中元素比较并插入。
代码中可以看到我开了滚动数组，但好像没什么必要，直接开二维就行。
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 3010
#define y(i , p) (f[i][p] + m * sum[i] * sum[i] + 2 * sum[n] * sum[i])
using namespace std;
typedef long long ll;
int q[N];
ll a[N] , sum[N] , f[N][2];
int main()
{
int n , m , i , j , l , r , d;
scanf("%d%d" , &n , &m);
for(i = 1 ; i <= n ; i ++ ) scanf("%lld" , &a[i]) , sum[i] = sum[i - 1] + a[i];
for(i = 1 ; i <= n ; i ++ ) f[i][1] = m * sum[i] * sum[i] - 2 * sum[n] * sum[i];
for(i = 2 ; i <= m ; i ++ )
{
l = r = 0 , d = i & 1;
for(j = 1 ; j <= n ; j ++ )
{
while(l < r && y(q[l + 1] , d ^ 1) - y(q[l] , d ^ 1) < 2 * m * sum[j] * (sum[q[l + 1]] - sum[q[l]])) l ++ ;
f[j][d] = y(q[l] , d ^ 1) - 2 * m * sum[j] * sum[q[l]] + m * sum[j] * sum[j] - 2 * sum[n] * sum[j];
while(l < r && (y(j , d ^ 1) - y(q[r] , d ^ 1)) * (sum[q[r]] - sum[q[r - 1]]) < (sum[j] - sum[q[r]]) * (y(q[r] , d ^ 1) - y(q[r - 1] , d ^ 1))) r -- ;
q[++r] = j;
}
}
printf("%lld\n" , f[n][m & 1] + sum[n] * sum[n]);
return 0;
}
标签：src cpp ace 证明 分享 while 开始 帮助 维护
原文地址：http://www.cnblogs.com/GXZlegend/p/6812435.html
