bzoj 3156 防御准备
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
斜率优化+DP
因为最后一个一定为守卫塔,所以输入时预处理到过来,接下来说的前i个其实是实际的后i个,反之亦然。
设f[i]为前i个检查点的最小费用,val[i]为在i处建塔的费用
假设在检查点i后,在j建塔比在k建塔费用更小(j>k之前均放置木偶)
f[j]=f[i]+(j-i-1)*(1+j-1-i)/2+val[j]<f[k]=f[i]+(k-i-1)*(1+k-1-i)/2+val[k]
即(j-i-1)*(j-i)/2+val[j]<(k-i-1)*(k-i)/2+val[k]
(j-i-1)*(j-i)-(k-i-1)*(k-i)<2*(val[k]-val[j])
(j-i)^2-j+i-(k-i)^2+k-i<2*(val[k]-val[j])
j^2-k^2-2*(j-k)*i+k-j<2*(val[k]-val[j])
(j^2-k^2+k-j-2*(val[k]-val[j]))/2*(j-k)<i
很好推吧,推出来了就出来了~~~~
#include<cstdio>
#include<cstring>
#include<iostream>
#define MAXN 1100000
using namespace std;
typedef long long ll;
ll n,head,tail;
ll q[MAXN],f[MAXN],val[MAXN],ans;
double get(ll k,ll j)
{
double p=(j*j-k*k-k+j-2.0*(f[k]-f[j]))/(2.0*(j-k));
return p;
}
int main()
{
scanf("%lld",&n);
for(int i=n;i>=1;i--)
scanf("%lld",&val[i]);
q[1]=1;f[1]=val[1];
head=tail=1ll;ans=val[1]+1ll*n*(n-1)/2ll;
for(int i=2;i<=n;i++)
{
while(head<tail&&get(q[head+1],q[head])<i)head++;
f[i]=f[q[head]]+1ll*(i-q[head])*(i-q[head]-
1)/2+val[i];
ans=min(ans,f[i]+1ll*(n-i)*(n-i+1)/2);
while(head<tail&&get(q[tail],q[tail-1])>get(i,q
[tail]))tail--;
q[++tail]=i;
}
printf("%lld",ans);
return 0;
}