玩具装箱TOY[斜率优化]

最新推荐文章于 2024-09-11 10:15:50 发布

原创最新推荐文章于 2024-09-11 10:15:50 发布 · 274 阅读

2 ·

CC 4.0 BY-SA版权

斜率优化专栏收录该内容

7 篇文章

订阅专栏

本文深入探讨了斜率优化动态规划技巧，通过实例讲解如何利用下凸包维护最小值，加速区间DP问题求解。文章提供了完整的代码实现，适合算法学习者及竞赛选手参考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门

第一次自己推斜率优化,好高兴

对于区间的长度,s[i]表示前缀和在加上i

也就是

$s[i]=i+\sum a[j](1<=j<=i)$

首先,考虑暴力的DP

$f[i] = min ( f[j] + (s[i]-s[j]-L-1)^2 )$

拆开

$f[i] = min ( f[j] + s[i]^2 + s[j]^2 + L^2 -2*s[i]*s[j] - 2*s[i]*L + 2*s[j]*L -2s[i] + 2s[j] + 2L +1)$

整理成如下形式

$f[i] = (f[j]+(...[j])) + (...)*(...[j]) +(...)$

也就是

$f[i] = min ( f[j] + s[j]^2 + 2*s[j] - 2*(s[i]-L)*s[j] + ...)$

最小值维护下凸包

#include<bits/stdc++.h>
#define y(A) (f[A]+s[A]*s[A]+s[A])
#define k(A) (2*s[A]-2*L)
#define x(A) (s[A])
#define N 50005
#define inf 1000000000000000
#define LL long long
using namespace std;
LL n,L,s[N],f[N],q[N];
double K(int i,int j){return 1.0*(y(j)-y(i))/(x(j)-x(i));}
int main(){
	scanf("%lld%lld",&n,&L);
	for(int i=1;i<=n;i++){
		LL x; scanf("%lld",&x);
		s[i]=s[i-1]+x;
	}
	for(int i=1;i<=n;i++) s[i]+=i;
	int l=1,r=1;
	for(int i=1;i<=n;i++){
		while(l<r && K(q[l],q[l+1]) <= k(i)) l++;
		f[i] = f[q[l]] + (s[i]-s[q[l]]-L-1)*(s[i]-s[q[l]]-L-1);
		while(l<r && K(q[r],i) <= K(q[r-1],q[r])) r--;
		q[++r] = i;
	}printf("%lld",f[n]); return 0;
}