[Bzoj1911][Apio2010]特别行动队(斜率优化)

题目链接

斜率优化的经典模型，将序列分成若干段，每段有一个权值计算方法，求权值和最大/小

暴力的dp $O(n^{2})$

dp[i]为1-i的序列的最优解。sum[i]为前缀和,$D(i)=ax^{2}+bx+c$

转移为$dp[i]=\max_{j=0}^{i-1}dp[j]+D(sum[i]-sum[j])$

然后叒是经典的推公式：

设$k<j<i$，且i从j转移比i从k转移更优。

$dp[j]+a(sum[i]-sum[j])^{2}-b(sum[i]-sum[j])+c\geq dp[k]+a(sum[i]-sum[k])^{2}-b(sum[i]-sum[k])+c$

$dp[j]+asum[j]^{2}-bsum[j]-(dp[k]+asum[k]^{2}-bsum[k])\geq 2asum[i](sum[j]-sum[k])$

$\frac{dp[j]+asum[j]^{2}-bsum[j]-(dp[k]+asum[k]^{2}-bsum[k])}{sum[j]-sum[k]}\geq 2asum[i]$

设   $f[j]=dp[j]+sum[j]^{2}-bsum[j]$   $T[j]=sum[j]$

$\frac{f[j]-f[k]}{T[j]-T[k]}\geq 2asum[i]$

将$(T[j],f[j])$,$(T[k],j[k])$看成二维平面上的点。所以当斜率$K_{jk}\geq 2asum[i]$时，从dp[j]转移比dpk]更优。

所以用优先队列维护最优的点集，就A了。

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const  int maxn = 1e6 + ;
 ll sum[maxn], q[maxn], dp[maxn], que[maxn];
 int n, a, b, c;
 ll f(int j, int k) {
     return dp[j] + a * sum[j] * sum[j] - b * sum[j] - (dp[k] + a * sum[k] * sum[k] - b * sum[k]);
 }
 ll T(int j, int k) {
     return sum[j] - sum[k];
 }
 ll D(int x) {
     return a * x*x + b * x + c;
 }
 int main() {
     scanf("%d", &n);
     scanf("%d%d%d", &a, &b, &c);
     for (int i = ; i <= n; i++)
         scanf("%lld", &q[i]), sum[i] = sum[i - ] + q[i];
     int l = , r = ;
     que[l] = ;
     for (int i = ; i <= n; i++) {
         while (l < r&&f(que[l + ], que[l]) >=  * a*sum[i] * T(que[l + ], que[l]))
             l++;
         dp[i] = dp[que[l]] + D(sum[i] - sum[que[l]]);
         while (l < r && f(que[r], que[r - ])*T(i, que[r]) < T(que[r], que[r - ])*f(i, que[r]))
             r--;
         que[++r] = i;
     }
     printf("%lld\n", dp[n]);
 }