【题目描述】
给出N个单词,每个单词有个非负权值Ci,现要将它们分成连续的若干段,每段的代价为此段单词的权值和,还要加一个常数M,即(∑Ci)^2+M。现在想求出一种最优方案,使得总费用之和最小。
【输入格式】
包含多组测试数据,对于每组测试数据。
第一行包含两个整数N和M(0 <= N <= 500000,0 <= M <= 10000)
第二行为N个整数(<=100000000)。
【输出格式】
输出仅一个整数,表示最小的价值。
斜率dp裸题,基础公式
考虑优化,设
则化简可得
维护队列即可
#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
using namespace std;
int n,head,tail;
long long sum[500005],f[500005];
int q[500005];
long long m;
double g(int k1,int k2,int i)
{
return double(f[k1]-f[k2]+sum[k1]*sum[k1]-sum[k2]*sum[k2])/double(sum[k1]-sum[k2]);
}
int main()
{
while(~scanf("%d%lld",&n,&m))
{
memset(f,0,sizeof(f));
for(int i=1;i<=n;i++) scanf("%lld",&sum[i]),sum[i]+=sum[i-1];
head=tail=0;
for(int i=1;i<=n;i++)
{
int k1=q[head],k2=q[head+1];
while(head<tail&&g(k1,k2,i)<=2*sum[i]) head++,k1=q[head],k2=q[head+1];
f[i]=f[k1]+(sum[i]-sum[k1])*(sum[i]-sum[k1])+m;
while(head<tail&&g(q[tail-1],q[tail],i)>=g(q[tail],i,i)) tail--;
q[++tail]=i;
}
printf("%lld\n",f[n]);
}
}