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

最新推荐文章于 2019-07-24 12:36:51 发布

原创最新推荐文章于 2019-07-24 12:36:51 发布 · 1.8k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#斜率优化 #题解 #bzoj

bzoj题解专栏收录该内容

84 篇文章

订阅专栏

本文介绍了一个关于防御准备的问题，通过动态规划及其斜率优化方法来解决该问题，给出了具体的算法实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【原题】

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

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;
}