HDU 3507 Print Article 斜率优化DP

题意：给出一篇文章打印每个字的花费，打印每行的花费为这一行每个字花费和的平方加m，求出将这篇文章打印到纸上的最低花费。

思路：如果是朴素的dp还是比较好想的，dp[i] = min{dp[j]+(sum[i]-sum[j])^2+m} (0<j<i),这是o(n^2)的算法，会超时，所以需要优化。

假设在算dp[i]时，k < j , 并且j点比k点更优，则：dp[j]+(sum[i]-sum[j])^2+m <= dp[k]+(sum[i]-sum[k])^2+m,

化简可得：dp[j]+sum[j]^2-(dp[k]+sum[k]^2) <= 2*sum[i]*(sum[j]-sum[k]);

令yj = dp[j]+sum[j]^2 , xj = sum[j];则上式可变为(yj-yk)/(xj-xk) <= 2*sum[i];

即说明若上式成立则j点比k点更优，令f[j,k] = (yj-yk)/(xj-xk);

可以得到2条结论：

1.若f[j,k] <= 2*sum[i] , 由于sum是递增的，则往后j点会一直比k点更优，则k点可以删除。

2.若f[j,k] > f[i,j] , 则j点可以扔掉，因为如果f[i,j] <= 2*sum[i],则i点比j点更优，j点可以扔掉，如果f[i,j]>2*sum[i],这必然f[j,k]>=2*sum[i],则k点比j点更优，j点可以删除。

这就相当于维护一个斜率上升的下凸包，用一个队列维护即可。

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <map>
#include <set>
#include <vector>
#include <cmath>
#include <stack>
#include <queue>
#include <cstdlib>
#include <algorithm>
using namespace std;
typedef __int64 int64;
typedef long long ll;
#define M 600005
#define N 1000005
#define max_inf 0x7f7f7f7f
#define min_inf 0x80808080
#define mod 1000000007
#define lc rt<<1
#define rc rt<<1|1

int n , m , head , tail;
int arr[M] , q[M] , sum[M] , dp[M];

bool Judge(int k , int j , int i)//判断j点是否比k点更优
{
	return dp[j]+sum[j]*sum[j]-(dp[k]+sum[k]*sum[k]) < 2*sum[i]*(sum[j]-sum[k]);
}

bool Cal(int k , int j , int i)//判断j点是否为上凸点
{
	int t1 = dp[j]+sum[j]*sum[j]-(dp[k]+sum[k]*sum[k]);
	int t2 = dp[i]+sum[i]*sum[i]-(dp[j]+sum[j]*sum[j]);
	return t1*(sum[i]-sum[j]) > t2*(sum[j]-sum[k]);
}

int main()
{
	int i;
	while (~scanf("%d%d",&n,&m))
	{
		sum[0] = head = tail = 0;
		dp[0] = q[tail++] = 0;
		for (i = 1 ; i <= n ; i++)
		{
			scanf("%d",arr+i);
			sum[i] = sum[i-1]+arr[i];
			while (tail-1 > head && Judge(q[head],q[head+1],i))head++;
			dp[i] = dp[q[head]]+(sum[i]-sum[q[head]])*(sum[i]-sum[q[head]])+m;

			while (tail-1 > head && Cal(q[tail-2],q[tail-1],i))tail--;
			q[tail++] = i;
		}
		printf("%d\n",dp[n]);
	}
	return 0;
}