Description
Kano想要打印一篇含有N个单词的文章,每个单词的打印费用是Ci,一行打印k个单词的费用是:
(∑i=1kCi)2+M其中M为定值。
Kano想知道打印这篇文章的最小费用。Sample Input & Output
5 5
5 9 5 7 5230
大概从这题开始下面都是类似的斜率优化裸题吧。
方程:
dp[i]=mink=1i−1{dp[k]+(∑j=k+1iCi)2+M}
g(x,y)=sum[y]2−sum[x]2+dp[y]−dp[x]2∗(sum[y]−sum[x])
#include <cstdio>
#include <cstring>
#include <algorithm>
#define M 500005
using namespace std;
int sum[M],dp[M],deq[M];
int sqr(int x){return x*x;}
int up(int x,int y){return (sqr(sum[y])-sqr(sum[x])+dp[y]-dp[x]);}
int down(int x,int y){return 2*(sum[y]-sum[x]);}
int main(){
int n,Mt;
while(scanf("%d %d",&n,&Mt)==2){
for(int i=1,x;i<=n;i++){
scanf("%d",&x);
sum[i]=sum[i-1]+x;
}
int L=0,R=-1; deq[++R]=0;
for(int i=1;i<=n;i++){
while(L<R&&up(deq[L],deq[L+1])<=sum[i]*down(deq[L],deq[L+1]))++L;
dp[i]=dp[deq[L]]+sqr(sum[i]-sum[deq[L]])+Mt;
while(L<R&&up(deq[R-1],deq[R])*down(deq[R],i)>=up(deq[R],i)*down(deq[R-1],deq[R]))--R;
deq[++R]=i;
}
printf("%d\n",dp[n]);
}
return 0;
}