题面:
蒟蒻的第一道斜率优化题目。
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。但他希望费用最小.
思路:
设dp[i]为前i个玩具全部放置好的最小费用,于是可以得到dp方程:
dp[i]=min(dp[j]+(s[i]−s[j]+i−j−L−1)2)
d
p
[
i
]
=
min
(
d
p
[
j
]
+
(
s
[
i
]
−
s
[
j
]
+
i
−
j
−
L
−
1
)
2
)
再化简一下:
f[i]=s[i]+if[j]=s[j]+jc=L+1dp[i]=min(dp[j]+(f[i]−f[j]−c)2)
f
[
i
]
=
s
[
i
]
+
i
f
[
j
]
=
s
[
j
]
+
j
c
=
L
+
1
d
p
[
i
]
=
min
(
d
p
[
j
]
+
(
f
[
i
]
−
f
[
j
]
−
c
)
2
)
然后可以把它化成斜率式:
dp[i]=dp[j]+f2[i]+(f[j]+c)2−2×f[i]×(f[j]+c)2f[i]×(f[j]+c)+(dp[i]−f2[i])=dp[j]+(f[j]−c)2
d
p
[
i
]
=
d
p
[
j
]
+
f
2
[
i
]
+
(
f
[
j
]
+
c
)
2
−
2
×
f
[
i
]
×
(
f
[
j
]
+
c
)
2
f
[
i
]
×
(
f
[
j
]
+
c
)
+
(
d
p
[
i
]
−
f
2
[
i
]
)
=
d
p
[
j
]
+
(
f
[
j
]
−
c
)
2
每一个决策点的坐标为 (dp[i]−f2[i],dp[j]+(f[j]−c)2) ( d p [ i ] − f 2 [ i ] , d p [ j ] + ( f [ j ] − c ) 2 )
于是我们就要在平面上维护这么一个下半截靠右的凸壳,不难发现依次添加的每一个点的横坐标递增,同时每一次转移的斜率也是递增的。
于是可以用单调队列来维护,即每一次加点的时候在凸壳后方弹掉无用的点,同时由于斜率单调递增,可以把前面斜率小于当前斜率的点给依次弹掉。
时间复杂度 Θ(n) Θ ( n ) 。
#include<bits/stdc++.h>
#define REP(i,a,b) for(int i=a,i##_end_=b;i<=i##_end_;++i)
typedef long long ll;
using namespace std;
void File(){
freopen("bzoj1010.in","r",stdin);
freopen("bzoj1010.out","w",stdout);
}
template<typename T>void read(T &_){
T __=0,mul=1; char ch=getchar();
while(!isdigit(ch)){
if(ch=='-')mul=-1;
ch=getchar();
}
while(isdigit(ch))__=(__<<1)+(__<<3)+(ch^'0'),ch=getchar();
_=__*mul;
}
const int maxn=5e4+10;
const int inf=0x3f3f3f3f;
int n,head,tail;
ll a[maxn],L,dp[maxn],f[maxn];
struct node{ll x,y;}qu[maxn];
long double getk(node b,node c){return 1.0*(c.y-b.y)/(c.x-b.x);}
node getn(int x){return (node){f[x]+L,dp[x]+(f[x]+L)*(f[x]+L)};}
void work(){
/*REP(i,1,n){
dp[i]=inf;
REP(j,0,i-1)dp[i]=min(dp[i],dp[j]+(f[i]-f[j]-L)*(f[i]-f[j]-L));
cout<<"num #"<<i<<" : "<<dp[i]<<endl;
}*/
head=1; tail=0;
qu[++tail]=getn(0);
REP(i,1,n){
while(head<tail && getk(qu[head],qu[head+1])<2*f[i])++head;
dp[i]=qu[head].y-2*f[i]*qu[head].x+f[i]*f[i];
//cout<<i<<" "<<dp[i]<<endl;
node nd=getn(i);
while(tail>head && getk(nd,qu[tail])<getk(qu[tail],qu[tail-1]))--tail;
qu[++tail]=nd;
}
printf("%lld\n",dp[n]);
}
void init(){
read(n); read(L);
REP(i,1,n)read(a[i]);
REP(i,1,n)f[i]=f[i-1]+a[i]+1;
L=L+1;
}
int main(){
File();
init();
work();
return 0;
}