斜率优化专题3——bzoj 3156 防御准备 题解

【原题】

3156: 防御准备

Time Limit: 10 Sec  Memory Limit: 512 MB

Submit: 198  Solved: 107

[Submit][Status]

Description

Input

第一行为一个整数N表示战线的总长度。

第二行N个整数，第i个整数表示在位置i放置守卫塔的花费Ai。

Output

共一个整数，表示最小的战线花费值。

Sample Input

10

2 3 1 5 4 5 6 3 1 2

Sample Output

18

HINT

1<=N<=10^6,1<=Ai<=10^9

Source

Katharon+#1

【分析】DP的斜率优化。

设f[1]=a[1]

方程是f[i]=min(f[j]+(i-j)*(i-j-1)/2+a[i])

---------------------------斜率优化---------------------------------

设状态K比状态J更优

f[k]+(i-k)*(i-k-1)/2+a[i]<f[j]+(i-j)*(i-j-1)/2+a[i]

设K=k+1,J=j+1

f[k]+(i-k)*(i-K)/2<f[j]+(i-j)*(i-J)

2*f[k]-i*k-i*K+k*K<2*f[j]-i*j-i*J+j*J

2*(f[k]-f[j])+k*K-j*J<i*(k+K-j-J)

(2*(f[k]-f[j])+k*K-j*J)/(k+K-j-J)<i

【代码】

#include<cstdio>
#include<algorithm>
using namespace std;
long long n,a[1000005],f[1000005],q[1000005],ans,h,t,i;
double xie(long long k,long long j)
{
return (2.0*(f[k]-f[j])+k*(k+1)-j*(j+1))/(2*k-2*j);
}
int main()
{
scanf("%lld",&n);
for (i=1;i<=n;i++)
scanf("%lld",&a[i]);
for (i=1;i<=n/2;i++) t=a[i],a[i]=a[n-i+1],a[n-i+1]=t;
f[1]=a[1];h=t=1ll;q[1]=1ll;
ans=a[1]+1ll*n*(n-1)/2ll;
for (i=2;i<=n;i++)
{
while (h<t&&xie(q[h+1],q[h])<i) h++;
f[i]=f[q[h]]+1ll*(i-q[h])*(i-q[h]-1)/2+a[i];
ans=min(ans,f[i]+1ll*(n-i+1)*(n-i)/2);
while (h<t&&xie(q[t],q[t-1])>xie(i,q[t])) t--;
q[++t]=i;
}
printf("%lld",ans);
return 0;
}