BZOJ3156: 防御准备 【斜率优化dp】

3156: 防御准备

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 2207  Solved: 933
[Submit][Status][Discuss]

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

练了几题，这类题目的模式还是很固定好写的

就不推了

设f[i]表示i位置放防御塔的最小代价

显然有f[i] = min{f[j] + (i - j) * (i - j - 1) / 2} + A[i]    【中间那一段就是中间木偶的代价】

去掉常数化简有2 * i * j + 2 * f[i] = (2 * f[j] + j^2 + j)

令y = 2 * f[j] + j^2 + j，x = j

就是y = 2i * x + 2 * f[i]

化为求截距最小，由于所有值都是单调递增，维护下凸包

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#define eps 1e-9#define LL long long int#define REP(i,n) for (int i = 1; i <= (n); i++)#define fo(i,x,y) for (int i = (x); i <= (y); i++)#define Redge(u) for (int k = head[u]; k != -1; k = edge[k].next)using namespace std;const int maxn = 1000005,maxm = 100005,INF = 1000000000;inline LL read(){LL out = 0,flag = 1;char c = getchar();while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}while (c >= 48 && c <= 57) {out = out * 10 + c - 48; c = getchar();}return out * flag;}LL n,q[maxn],head,tail;LL A[maxn],f[maxn];inline double slope(LL u,LL v){double y1 = 2 * f[u] + u * u + u,y2 = 2 * f[v] + v * v + v;return (y1 - y2) / (u - v);}inline LL getf(LL i,LL j){return f[j] + (i - j) * (i - j - 1) / 2 + A[i];}int main(){n = read();REP(i,n) A[i] = read();head = tail = 0;for (LL i = 1; i <= n; i++){while (head < tail && slope(q[head],q[head + 1]) + eps < 2 * i) head++;f[i] = getf(i,q[head]);while (head < tail && slope(q[tail],q[tail - 1]) + eps > slope(i,q[tail])) tail--;q[++tail] = i;}cout<<f[n]<<endl;return 0;}