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，f[i]表示i点放置守卫塔，1~i的最小代价。

那么f[i]=min{f[j]+a[i]+【(j+1)~(i-1)与i的距离和】}

其中，距离和可以通过简单的数学式子表示出来：

i-j-1+i-j-2+……+1

就是个等差数列，也就是

(i-j)*(i-j-1)/2

所以f[i]=min{f[j]+a[i]+(i-j)*(i-j-1)/2}

接下来就是斜率优化的过程……

推出来的斜率似乎是(2(f[j]-f[k])+j*j+j-k*k-k)/(j-k)

推得仔细些……别推错了……= =

对了要long long

#include<bits/stdc++.h>#define ll long longusing namespace std;const int N=1000005;int n,Q[N];ll a[N],f[N];ll ANS(int i,int j){return f[j]+(ll)(((i-j)*(i-j-1))>>1LL)+a[i];}bool ok(ll i,ll j,ll k){ll t1=2LL*(f[i]-f[j])+i*i+i-j*j-j,t2=i-j,t3=2LL*(f[j]-f[k])+j*j+j-k*k-k,t4=j-k;return t1*t4>t3*t2;}int main(){scanf("%d",&n);for (int i=1;i<=n;i++) scanf("%lld",&a[i]);int head=1,tail=1;Q[1]=0;for (int i=1;i<=n;i++){while (head<tail && ANS(i,Q[head])>ANS(i,Q[head+1])) head++;f[i]=ANS(i,Q[head]);while (head<tail && ok(Q[tail-1],Q[tail],i)) tail--;Q[++tail]=i;}printf("%lld\n",f[n]);return 0;}