【bzoj1911】[Apio2010]特别行动队commando

最新推荐文章于 2017-08-05 21:27:09 发布

KikiDMW

最新推荐文章于 2017-08-05 21:27:09 发布

阅读量403

点赞数

分类专栏：动态规划斜率优化

本文链接：https://blog.csdn.net/KikiDMW/article/details/69950780

版权

动态规划同时被 2 个专栏收录

28 篇文章 0 订阅

订阅专栏

斜率优化

3 篇文章 0 订阅

订阅专栏

题目链接

Description

这里写图片描述

Input

这里写图片描述

Output

这里写图片描述

Sample Input

4
-1 10 -20
2 2 3 4

Sample Output

HINT

这里写图片描述

题解

~~斜率优化好像也没有什么特殊的技巧哇。~~
设 $s_i$ 为 $1\dots i$ 的和， $f_i$ 为 $1\dots i$ 的最大费用。
那么就愉快的推式子吧：
首先：

f i = m a x {f j + a (s i - s j) 2 + b (s i - s j) + c}

$f_i=max\{f_j+a(s_i-s_j)^2+b(s_i-s_j)+c\}$
设

k<j $k<j$ 那么由j转移更优的话满足：

f j + a (s i - s j) 2 + b (s i - s j) + c > f k + a (s i - s k) 2 + b (s i - s k) + c

$f_j+a(s_i-s_j)^2+b(s_i-s_j)+c>f_k+a(s_i-s_k)^2+b(s_i-s_k)+c$

\Rightarrow s i > f k - f j + b s j - b s k + a s 2 k - a s 2 j 2 a s k - 2 a s j

$\Rightarrow s_i>\frac{f_k-f_j+bs_j-bs_k+as_k^2-as_j^2}{2as_k-2as_j}$
然后就是斜率优化经典的那一套理论啦。

#include<bits/stdc++.h>
using namespace std;

inline int read(){
    int x = 0, f = 1; char c = getchar();
    while(!isdigit(c)) { if(c == '-') f = -1; c = getchar(); }
    while(isdigit(c)) { x = x * 10 + c - '0'; c = getchar(); }
    return x * f;
}

typedef long long ll;

const int N = 1000000 + 10;
ll s[N], f[N], a, b, c;
int n, q[N];

void init(){
    n = read();
    a = read(), b = read(), c =read();
    int x;
    for(int i = 1; i <= n; i++) x = read(), s[i] = s[i-1] + x;
}

inline double slope(int k, int j){
    return (double)((f[k]-f[j]+b*s[j]-b*s[k]+a*s[k]*s[k]-a*s[j]*s[j])/(2*a*s[k]-2*a*s[j]));
}
inline ll sqr(ll x){return x * x;}

void work(){
    int l = 0, r = 0;
    for(int i = 1; i <= n; i++){
        while(l < r && s[i] > slope(q[l], q[l+1])) l++;
        int j = q[l];
        f[i] = f[j] + a * sqr(s[i] - s[j]) + b * (s[i] - s[j]) + c;
        while(l < r && slope(q[r], i) < slope(q[r-1], q[r])) r--;
        q[++r] = i;
    }
    printf("%lld\n", f[n]);
}

int main(){
    init();
    work();
    return 0;
}