BZOJ 1010 [HNOI2008]玩具装箱toy (斜率优化DP)

1010: [HNOI2008]玩具装箱toy

Time Limit: 1 Sec   Memory Limit: 162 MB
Submit: 11388   Solved: 4801
[ Submit][ Status][ Discuss]

Description

　　P教授要去看奥运，但是他舍不下他的玩具，于是他决定把所有的玩具运到北京。他使用自己的压缩器进行压
缩，其可以将任意物品变成一堆，再放到一种特殊的一维容器中。P教授有编号为1...N的N件玩具，第i件玩具经过
压缩后变成一维长度为Ci.为了方便整理，P教授要求在一个一维容器中的玩具编号是连续的。同时如果一个一维容
器中有多个玩具，那么两件玩具之间要加入一个单位长度的填充物，形式地说如果将第i件玩具到第j个玩具放到一
个容器中，那么容器的长度将为 x=j-i+Sigma(Ck) i<=K<=j 制作容器的费用与容器的长度有关，根据教授研究，
如果容器长度为x,其制作费用为(X-L)^2.其中L是一个常量。P教授不关心容器的数目，他可以制作出任意长度的容
器，甚至超过L。但他希望费用最小.

Input

　　第一行输入两个整数N，L.接下来N行输入Ci.1<=N<=50000,1<=L,Ci<=10^7

Output

　　输出最小费用

Sample Input

5 4
3
4
2
1
4

Sample Output

       1

了解斜率DP之后，信心满满的来A这道题，可怕在WA两位数之后，参考别人的写法过了，后来再修改，还是WA，经过无数次折磨之后才知道是最后入队列的问题，应该是++i，我一直在执行i++，然后花样WA。

通过此题对斜率优化 DP 有了更深层次的理解，但是化简的过程实在是折磨人，而且竟然还化简出了不能进行斜率DP的dp[i]*dp[j]形式，要不是知道这个题是斜率DP，估计就放弃了。推断的过程都是千篇一律。

看到最小费用就想用最小费用流，虽然不怎么会用。哎。。。

代码实现：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<cstdio>
#define ll long long
#define mset(a,x) memset(a,x,sizeof(a))

using namespace std;
const double PI=acos(-1);
const int inf=0x3f3f3f3f;
const double esp=1e-6;
const int maxn=5e4+5;
const int mod=1e9+7;
int dir[4][2]={0,1,1,0,0,-1,-1,0};
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
ll inv(ll b){if(b==1)return 1; return (mod-mod/b)*inv(mod%b)%mod;}
ll fpow(ll n,ll k){ll r=1;for(;k;k>>=1){if(k&1)r=r*n%mod;n=n*n%mod;}return r;}
ll n,m,q[maxn],head,tail;
ll sum[maxn],dp[maxn];

double get_k(ll y,ll x)
{
	return ((dp[x]-dp[y]+(sum[x]+m)*(sum[x]+m)-(sum[y]+m)*(sum[y]+m))/(2.0*(sum[x]-sum[y])));
}

void DP()
{
	ll i,j,k;
	tail=1;head=1;
	q[tail]=0;
	for(i=1;i<=n;i++)
	{
		while(head<tail&&get_k(q[head],q[head+1])<=sum[i])
		head++;
		
		dp[i]=dp[q[head]]+(sum[i]-sum[q[head]]-m)*(sum[i]-sum[q[head]]-m);
		
		while(head<tail&&get_k(q[tail],i)<get_k(q[tail-1],q[tail]))
		tail--;
		
		q[++tail]=i;
	}
}

int main()
{
	ll i,j,k,x;
	while(cin>>n>>m)
	{
		m++;
		mset(dp,0);
		mset(q,0);
		mset(sum,0);
		for(i=1;i<=n;i++)
		{
			cin>>x;
			sum[i]=sum[i-1]+x;
		}
		for(i=1;i<=n;i++)
		sum[i]+=i;
		
		DP();
		cout<<dp[n]<<endl;
	}
	return 0;
}