Print Article(HDU 3507 -- 斜率dp)

题目来源：http://acm.hdu.edu.cn/showproblem.php?pid=3507

---

题意

将$n$个数拆分成若干段，每段的打印代价为$m+ \sum_{i=1}^k c[i]$，求最小代价

分析

$dp[i]$ : 打印前 $i$ 页的最小代价
可以得到一个时间复杂度为$O(n*n)$的转移方程

$$dp[i] = min \{ dp[j] + (sum[i]-sum[j])^2+m \}$$

假设$k \lt j \lt i$，若选择以$j$为最优转移方案，则有

$$dp[j] + (sum[i]-sum[j])^2+m \le dp[k] + (sum[i]-sum[k])^2+m$$
对上列不等式子化简得：

$$\frac{(dp[j]+sum[j]^2)-(dp[k]+sum[k]^2)}{sum[j]-sum[k]} \le 2*sum[i]$$
令 $y_i = dp[i]+sum[i]^2$， $x_i = sum[i]$，则上式为
$$\frac{y_j - y_k}{x_j-x_k} \le 2*sum[i]$$
不等号左边是一个 $(j,k)$的斜率，记为 $K_{j,k}$

至此，我们可以知道第二维的时间可以通过斜率优化，降为$O(1)$，具体的做法是：

维护一个单调队列，使得 $K_{q[u],q[u+1]} \le K_{q[u+1],q[u+2]}$（因为，当$K_{q[u],q[u+1]} \gt K_{q[u+1],q[u+2]}$ 时，$q[u+1]$是无效点，可以从图像上理解一下）
显然，只要从队首往后与相邻比较，如果有

$$\frac{y_{q[L+1]} - y_{q[L]}}{x_{q[L+1]}-x_{q[L]}} \le 2*sum[i]$$
那么 $q[L+1]$ 必然优于 $q[L]$，此时舍弃 $q[L]$（因为，当 $q[u]$ 不优于 $q[v]$， $q[v]$ 不优于 $q[w]$，那么 $q[u]$ 必然不优于 $q[w]$，保留无意义）

---

代码

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<algorithm>
#include<time.h>
#include<iostream>
#include<vector>
#include<stack>
#include<queue>
#include<set>
#include<map>

using namespace std;

#define bll long long

const int maxn = 5e5+100;
int n,m,c,q[maxn];
long long dp[maxn],sum[maxn];

long long calup(int j,int k)
{
    return dp[j]+sum[j]*sum[j]-dp[k]-sum[k]*sum[k];
}

long long caldown(int j,int k)
{
    return sum[j]-sum[k];
}

int main()
{
    while (scanf("%d %d",&n,&m)!=EOF)
    {
        sum[0] = 0;
        for (int i=1;i<=n;i++) 
        {
            scanf("%d",&c);
            sum[i] = sum[i-1]+c;
        }

        int L = 0, R = 1;
        dp[0] = q[0] = 0;
        for (int i=1;i<=n;i++)
        {
            while (R>L+1 && calup(q[L+1],q[L])<=caldown(q[L+1],q[L])*sum[i]*2) L++;

            int j = q[L];
            dp[i] = dp[j]+(sum[i]-sum[j])*(sum[i]-sum[j])+m;

            while (R-2>=0 && calup(i,q[R-1])*caldown(q[R-1],q[R-2]) <= calup(q[R-1],q[R-2])*caldown(i,q[R-1])) R--;
            q[R++] = i;
        }
        printf("%lld\n",dp[n]);
    }
    return 0;
}