题目描述:
Description
P教授要去看奥运,但是他舍不下他的玩具,于是他决定把所有的玩具运到北京。他使用自己的压缩器进行压缩,其可以将任意物品变成一堆,再放到一种特殊的一维容器中。P教授有编号为1…N的N件玩具,第i件玩具经过压缩后变成一维长度为Ci.为了方便整理,P教授要求在一个一维容器中的玩具编号是连续的。同时如果一个一维容器中有多个玩具,那么两件玩具之间要加入一个单位长度的填充物,形式地说如果将第i件玩具到第j个玩具放到一个容器中,那么容器的长度将为 x=j-i+Sigma(Ck) i<=K<=j 制作容器的费用与容器的长度有关,根据教授研究,如果容器长度为x,其制作费用为(X-L)^2.其中L是一个常量。P教授不关心容器的数目,他可以制作出任意长度的容器,甚至超过L。但他希望费用最小.
Input
第一行输入两个整数N,L.接下来N行输入Ci.1<=N<=50000,1<=L,Ci<=10^7
Output
输出最小费用
Sample Input
5 4
3
4
2
1
4
Sample Output
1
题解:
这题显然可以推出DP式来,f[i]=min(f[j]+(s[i]-s[j]+i-j-1-L)^2)。
显然,这样做的复杂度是n^2,对于这题是难以接受的。
这里就要引入斜率优化的概念了。斜率优化是对于一类形如f[i]=min(f[j]+(a[i]-a[j])^2)这种转移方程和i和j都有关系的DP。
我们可以先从数字的方式理解斜率优化:K神的blog
再从其“形”来理解:方指导的blog
其实总而言之,就是要推出一个斜率的式子,如:(Y(j)−Y(k))/(X(j)−X(k))<=A[i], 其中A必须是递增的数组。这样我们就可以通过单调队列,以k神blog中所讲的方式开始处理了,复杂度可以达到O(n)。
对于本题而言,可以从f[j]+(s[i]-s[j]+i-j-1-L)^2>=f[k]+(s[i]-s[k]+i-j-1-L)^2(i>j>k>0)入手,推出斜率的关系式。这个式子就是:(f[k]+(sum[k]+L+1)^2-dp[j]-(sum[j]+L+1)^2)/2*(sum[k]-sum[j])(为了方便推式子,sum[i]表示前i个玩具的长度之和+i)
接下来就不难思考了。
代码如下:
#include<cstdio>
#include<string>
using namespace std;
const int maxn=50005;
int n,m,a[maxn],que[maxn];
long long f[maxn],s[maxn];
inline int read(){
int x=0; char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
return x;
}
double xl(int j,int k) {return (f[k]+(s[k]+m)*(s[k]+m)-f[j]-(s[j]+m)*(s[j]+m))/(2.0*(s[k]-s[j]));}
int main(){
n=read(),m=read(); m++;
for (int i=1;i<=n;i++) a[i]=read(),s[i]=s[i-1]+a[i];
for (int i=1;i<=n;i++) s[i]+=i;
int l=1,r=1; que[r]=0;
for (int i=1;i<=n;i++){
while (l<r&&xl(que[l],que[l+1])<=s[i]) l++;
f[i]=f[que[l]]+(s[i]-s[que[l]]-m)*(s[i]-s[que[l]]-m);
while (l<r&&xl(que[r],i)<xl(que[r-1],que[r])) r--;
que[++r]=i;
}
printf("%lld\n",f[n]);
return 0;
}