BZOJ 1010: [HNOI2008]玩具装箱toy
题意概述:
给N个物品,长度为c,可将连续的一段物品分到一起,令x=j-i+sigma(c[k])(i<=k<=j),则花费代价为(x-L)^2 (L为常量),如何分组使其费用最小.
题目分析:
1.一开始可以思考一下朴素的DP,写出其转移方程,如下:
ans[i]表示到i位置的最小费用,sum[i]表示1到i的长度和
令len[i]=sum[i]+i,
一次转移的复杂度为O(n),总的时间复杂度为O(n^2),但是1<=N<=50000,
O(n^2)方法显然要超时,观察能否进行优化.
2.进行对式子的推导:
假设当前i号的决策,存在j比k更优(j < k),则有
将带有i的项放在右边
因为j < k,则有len[j]-len[k] < 0,所以
定义
则若slope(j,k) > len[i],j比k更优,O(1)判断.
3.在此基础上,维护一个斜率递增的单调队列即可实现优化.
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=50000+10;
int que[maxn],N,L;
ll len[maxn],ans[maxn];
double slope(int a,int b)//a<b
{
return (ans[a]-ans[b]+(len[a]+L+1)*(len[a]+L+1)-(len[b]+L+1)*(len[b]+L+1))/(2.0*(len[a]-len[b]));
}
int main()
{
scanf("%d%d",&N,&L);
for(int l,i=1;i<=N;i++) {
scanf("%d",&l);
len[i]=len[i-1]+l;
}
for(int i=1;i<=N;i++) len[i]+=i;
int head=0,tail=0;
for(int i=1;i<=N;i++) {
while(head<tail&&slope(que[head],que[head+1])<len[i]) head++;
ans[i]=ans[que[head]]+(ll)(len[i]-len[que[head]]-L-1)*(len[i]-len[que[head]]-L-1);
while(head<tail&&slope(que[tail-1],que[tail])>slope(que[tail],i)) tail--;
que[++tail]=i;
}
printf("%lld",ans[N]);
return 0;
}