dp优化———斜率优化（复习）

最新推荐文章于 2020-12-10 23:21:47 发布

BIT_jzx

最新推荐文章于 2020-12-10 23:21:47 发布

阅读量122

点赞数

分类专栏：单调队列 dp 动态规划斜率优化总结文章标签：复习总结

本文链接：https://blog.csdn.net/weixin_43823476/article/details/99125612

版权

总结同时被 3 个专栏收录

43 篇文章 0 订阅

订阅专栏

dp 动态规划

27 篇文章 0 订阅

订阅专栏

单调队列

8 篇文章 0 订阅

订阅专栏

自己dp是学得真的差

一.复习

首先用例题复习：Print Article

首先可以很简单地列出dp式： $dp[i] = min( dp[j] + M + (sum[i] - sum[j])^{2}$

然后就可以用斜率套了

首先我们自己定义有 j < k

且 $dp[j] + M + (sum[i] - sum[j])^{2}<dp[k] + M + (sum[i] - sum[k])^{2}$

经过一系列移项后就可以得到：

$dp[j] + sum[j]^{2} - 2 * sum[i] * sum[j] < dp[k] + sum[k]^{2} - 2 * sum[i] * sum[k]$

我们设 $y_{j} = dp[j] + sum[j]^{2} ,x_{j} = 2*sum[j]$

这样就变成了 $y_{j} - y_{k}/ x_{j} - x_{k} > sum[i]$

这就是求斜率，所以就可以用单调队列了

注意要变成乘法。

这里有一个检验自己dp式是否列对的方法：要化成以上的形式，且不等式右边的常数一定是有单调性的（多道题的经验）

开始贴代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
using namespace std;
const int MAXN = 500003;
int n , m , dp[MAXN] , que[MAXN] , head , tail , sum[MAXN] , a[MAXN];
int qy( int i ){
    return dp[i] + sum[i] * sum[i];
}
int main()
{
    while( ~scanf( "%d%d" , &n , &m ) ){
        sum[0] = 0;
        head = tail = 0;
        memset( dp , 0 , sizeof( dp ) );memset( que , 0 , sizeof( que ) );
        for( int i = 1 ; i <= n ; i ++ ){
            scanf( "%d" , &a[i] );
            sum[i] = sum[i-1] + a[i];
        }
        head = 1 , tail = 1;
        que[1] = 0;
        for( int i = 1 ; i <= n ; i ++ ){
            while( head < tail && qy( que[head + 1] ) - qy( que[head] ) < 2 * sum[i] *( sum[que[head+1]]-sum[que[head]])  )
                head ++;
            dp[i] = ( sum[i] - sum[que[head]] ) * ( sum[i] - sum[que[head]] ) + dp[que[head]] + m ;
            while( head < tail && ( qy( i ) - qy( que[tail] ) ) * ( sum[que[tail]] - sum[que[tail-1]]) <= ( qy( que[tail] ) - qy( que[tail-1] ) ) * ( sum[i] - sum[que[tail]]) )
                tail --;
            que[++tail] = i;
        }
        printf( "%d\n" , dp[n] );
    }
    return 0;
}