首页 > 代码库 > 斜率优化专题3——bzoj 3156 防御准备 题解

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

2024-07-03 04:20:21   223人阅读

【原题】

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


联系
我们