【训练题62：dp + 斜率优化】Print Article | HDU3507

题意

• Print Article | HDU3507
  有一个长度为 $n$ 的序列 $C$
  把它分成一些子区间，总的花费为：
  $$\sum _{每个子区间k}\left(\right(\sum _{i在子区间k}{C}_i{)}^2+M)$$
  求总花费的最小值
• $1\le n\le 5\times 10^5$
  $1\le C_i\le 10^3$

思路

• 我们直接进行一个 $DNA$ 的动，设 $dp[i]$ 表示选完了 $1\sim i$ 的位置，最小的总花费
  记录一个前缀和数组 $pre[x]={\sum }_{i=1}^x{C}_i$
  转移也很简单：
  d p [ i ] = min ⁡ j = 0 i − 1 { d p [ j ] + ( p r e [ i ] − p r e [ j ] ) 2 } + M dp[i]=\min_{j=0}^{i-1}\{ dp[j]+(pre[i]-pre[j])^2 \} + M dp[i]=j=0mini−1​{dp[j]+(pre[i]−pre[j])2}+M
  但是这个复杂度很明显为 $O(n^2)$ ，我们怎么去优化呢？
  根据决策单调性的相关知识，我们可以通过某种方式将其优化！
• 设 $0\le k_2<k_1<i$ ，假设我们从 $k_1$ 那边转移到 $i$ 更优，那么我们有：
  $$dp[k_2]+(pre[i]-pre[k_2]{)}^2<dp[k_1]+(pre[i]-pre[k_1]{)}^2$$
• 化简得到：
  $$(dp[k_2]-dp[k_1]+pre[k_2{]}^2-pre[k_1{]}^2)/(pre[k_2]-pre[k_1])>2\times pre[i]$$
  我们令 $Y(x)=dp[x]+pre[x{]}^2$
  令 $X(x)=pre[x]$
  那么式子就变成了：
  $$\frac{Y(k_2)-Y(k_1)}{X(k_2)-X(k_1)}>2\times pre[i]$$
• 而右边我们放的是一个常数，假设为 $K(i)$ ，那式子可以想成：其中两个点之间的斜率大于 $K(i)$
  因为我们一开始定义 $k_1$ 更优的话，要求有这个式子
  我们可以记录，从这个点 $k_1$ 出发，到之后的某个 $k_2$ ，斜率最小是多少？ 如果斜率最小都 $>K(i)$ 了，自然选这个点是最最优的。
  我们可以使用求凸包的方法，快速去维护。
  注意到，随着 $i$ 递增，$pre[i]$ 也是单递增的，也就是我们的 $K(i)$ 也是单递增 的，我们就可以用类似单调队列 的写法，达到 $O(N)$ 的优秀复杂度

更套路的做法

• 主要是上面的转移，希望写得更快一点 （???） ，有新的办法去做
  考虑式子：
  d p [ i ] = min ⁡ j = 0 i − 1 { d p [ j ] + ( p r e [ i ] − p r e [ j ] ) 2 } + M dp[i]=\min_{j=0}^{i-1}\{ dp[j]+(pre[i]-pre[j])^2 \} + M dp[i]=j=0mini−1​{dp[j]+(pre[i]−pre[j])2}+M
• 我们省略 $\min$ 函数，拆开来变成：
  $$dp[i]=dp[j]+pre[i{]}^2+pre[j{]}^2-2pre[i]pre[j]+M$$
  由于考虑斜率，我们希望写成 $y=kx+b$ 的形式：
  $$dp[j]+pre[j{]}^2=2pre[i]pre[j]+(dp[i]-pre[i{]}^2-M)$$
  这里，只与 $j$ 有关的函数我们放在 $y$ 的位置
  与 $i,j$ 都有关的函数我们放在 $kx$ 的位置，并令 $k$ 表示只与 $i$ 有关的内容，令 $x$ 表示只与 $j$ 有关的内容
  只与 $i$ 有关的函数我们放在 $b$ 的位置
  然后我们发现，其实就是之前的 $Y(x)=dp[x]+pre[x{]}^2$ ， $X(x)=pre[x]$ ，$K(x)=2pre[x]$
• 接下来，我们需要求 $dp[i]$ 最小，就是希望我们 $b$ 最小；由于我们的 $x,k$ 都是单递增的，所以我们构造 下凸包，就可以达到我们的目的
  
• 我这里给出一个稍微封装的代码，就非常利于理解了吧

代码

• 时间复杂度：$O(N)$

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
using namespace std;
typedef long long ll;
void show(){std::cerr << endl;}template<typename T,typename... Args>void show(T x,Args... args){std::cerr << "[ " << x <<  " ] , ";show(args...);}

const int MAX = 5e5+50;
const int MOD = 1e9+7;
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const double EPS = 1e-5;

ll pre[MAX];
ll dp[MAX];
int M;
inline ll X(int x){return pre[x];}				   // 修改这里
inline ll Y(int x){return dp[x] + pre[x] * pre[x];}	// 修改这里
inline ll K(int x){return 2LL * pre[x];}		   // 修改这里
inline ll DX(int a,int b){return X(a) - X(b);}
inline ll DY(int a,int b){return Y(a) - Y(b);}
inline void F(int i,int j){						  // 修改这里
    dp[i] = dp[j] + (pre[i] - pre[j]) * (pre[i] - pre[j]) + M;
}
int Q[MAX],head,rear;
int main()
{
    int n;
    while(n = read()){
        M = read();
        for(int i = 1;i <= n;++i){
            int t = read();
            pre[i] = pre[i-1] + t;
            dp[i] = 0;
        }
        head = rear = 1;
        Q[head] = 0;dp[0] = 0;
        for(int i = 1;i <= n;++i){		// 注意符号方向 这是一个下凸包
            while(head < rear && DY(Q[head],Q[head+1]) >= K(i) * DX(Q[head],Q[head+1]))++head;
            int j = Q[head];
            F(i,j);
            while(head < rear && DY(Q[rear-1],Q[rear])* DX(Q[rear],i) >= DY(Q[rear],i) * DX(Q[rear-1],Q[rear]))--rear;
            Q[++rear] = i;
        }
        cout << dp[n] << endl;
    }

    return 0;
}