玩具装箱toy
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
斜率优化的入门题……
然而当时却没有去写……
然后就因为斜率优化写得不够熟而被同学出的题给坑了……
(╯‵□′)╯︵┻━┻
思路:
可以很快想到一个很显然的
O(n2)
的dp:
dp[i]=max{dp[j]+(c[i]+i−c[j]−j−1−l)2}
然后咱试图去优化它。
首先,发现c数组的值总是和对应的i共存,那么咱令
f[i]=c[i]+i
,
L=l+1
,
那么
dp[i]=max{dp[j]+(f[i]−f[j]+L)2}
。
通过展开可以证明上式具有单调性。
(其实有一个通用的技巧是,不去证单调性而是写好后和暴力对拍观察有无单调性)
于是考虑斜率优化:
设在位置i,k的决策比j优。
那么
dp[k]+(f[i]−f[k]+L)2>dp[j]+(f[i]−f[j]+L)2
把
f[i]+L
视作一项,对上式进行展开并化简得:
dp[k]−2∗(f[i]+L)∗f[k]+f[k]2>dp[j]−2∗(f[i]+L)∗f[j]+f[j]2
整理得:
dp[j]−dp[k]−2∗L∗(f[j]−f[k])+f[j]2−f[k]2<2∗f[i]∗(f[j]−f[k])
给左半边配个方:
dp[j]−dp[k]+(f[j]+L)2−(f[k]+L)2<2∗f[i]∗(f[j]−f[k])
把它写成斜率的形式:
dp[j]−dp[k]+(f[j]+L)2−(f[k]+L)22∗(f[j]−f[k])<f[i]
于是咱得到了一个可以 O(1) 判断最优解的式子,咱称他为 slope(i,j) 。
当j < k且上式被满足,则k比j更优。
由于决策单调性,可以维护一个单调队列,做到
O(1)
转移:
当
slope(q[l],q[l+1])<f[i]
,
q[l+1]
更优,弹出
q[l]
;
当
slope(q[r−1],q[r])>slope(q[r],i)
,则留下的
q[r]
必会在
q[r−1]
被弹出的同时弹出,为无效状态,于是弹出
q[r]
。
每次从队首获取最优决策即可~
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef double db;
inline long long read()
{
long long x=0;char ch=getchar();
while(ch<'0' || '9'<ch)ch=getchar();
while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
return x;
}
const int N=50009;
ll n,l,p;
ll dp[N],f[N],c[N];
ll q[N],ls,r;
inline db sqr(ll a)
{
return (db)a*a;
}
inline db slope(int j,int k)
{
return ((db)dp[k]-dp[j]+sqr(f[k]+l)-sqr(f[j]+l))/(2.0*(f[k]-f[j]));
}
int main()
{
n=read();
l=read()+1;
for(int i=1;i<=n;i++)
{
c[i]=read()+c[i-1];
f[i]=c[i]+i;
}
ls=1,r=0;
q[++r]=0;
for(int i=1;i<=n;i++)
{
while(ls<r && slope(q[ls],q[ls+1])<=f[i])
ls++;
dp[i]=dp[q[ls]]+(ll)sqr(f[i]-f[q[ls]]-l);
while(ls<r && slope(q[r],i)<slope(q[r-1],q[r]))
r--;
q[++r]=i;
}
printf("%lld\n",dp[n]);
return 0;
}