斜率优化 dp 总结

我们以一道例题引入:

洛谷 P2365 任务安排:


(n) 个任务排成一个序列在一台机器上等待完成(顺序不得改变),这 (n) 个任务被分成若干批,每批包含相邻的若干任务。

从零时刻开始,这些任务被分批加工,第 (i) 个任务单独完成所需的时间为 (t_i)​。在每批任务开始前,机器需要启动时间 (s),而完成这批任务所需的时间是各个任务需要时间的总和(同一批任务将在同一时刻完成)。

每个任务的费用是它的完成时刻乘以一个费用系数 (C_i)​。请确定一个分组方案,使得总费用最小。

(dp_i) 为选取到第 (i) 个任务时的最大价值,枚举一个起点 (j) 分批可以得到:

[dp_i=min{dp_i,dp_j+ ext{sum}_T(1,i) ext{sum}_C(i,j+1)+scdot ext{sum}_C(n,j+1)} ]

其中对于序列 (S)

[ ext{sum}_S(l,r)=sum_{i=l}^rS_i= ext{sumS}_r- ext{sumS}_{l-1} ]

后面这个等号是求法,( m sumS) 是前缀和,也就是

[ ext{sumS}_i=sum_{k=1}^iS_i ]

时间复杂度 (O(n^2)) .

Code:

#include<ctime>
#include<queue>
#include<stack>
#include<cmath>
#include<iterator>
#include<cctype>
#include<vector>
#include<map>
#include<set>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<bitset>
using namespace std;
const int N=5005;
int n,s,t[N],c[N],sumT[N],sumC[N],dp[N];
int main()
{
	scanf("%d%d",&n,&s);
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d",t+i,c+i);
		sumT[i]=sumT[i-1]+t[i]; sumC[i]=sumC[i-1]+c[i];
	} memset(dp,0x3f,sizeof dp); dp[0]=0;
	for (int i=1;i<=n;i++)
		for (int j=0;j<i;j++)
			dp[i]=min(dp[i],dp[j]+sumT[i]*(sumC[i]-sumC[j])+s*(sumC[n]-sumC[j]));
	printf("%d",dp[n]);
	return 0;
}

这个 (O(n^2)) 的做法还是太慢了,过不了 (3 imes 10^5) 的数据,考虑优化。

要优化,当然先对动态转移方程变形:

[egin{aligned}dp_i&=min{dp_i,dp_j+ ext{sum}_T(1,i) ext{sum}_C(i,j+1)+scdot ext{sum}_C(n,j+1)}\&=dp_j-( ext{sumT}_i+s) ext{sumC}_j+ ext{sumT}_i ext{sumC}_i+scdot ext{sumC}_n& ext{设 }j ext{ 是使得 }dp_i ext{ 取到最小值的 }j\Rightarrow dp_j&=( ext{sumT}_i+s) ext{sumC}_j+dp_i- ext{sumT}_i ext{sumC}_i-scdot ext{sumC}_nend{aligned} ]

此时令 (egin{cases}dp_j=y& ext{因变量}\ ext{sumT}_i+s=k& ext{斜率}\ ext{sumC}_j=x& ext{自变量}\dp_i- ext{sumT}_i ext{sumC}_i-scdot ext{sumC}_n=b& ext{截距}end{cases}) 可以得到:

[y=kx+b ]

发现这正好是一个直线方程,并且 (x) 是随着 (j) 改变而改变的。

别忘了我们的目标:使 (dp_i) 最小,要使 (dp_i) 最小,就要让 (b)(截距)最小((b) 中除了 (dp_i) 以外的东西都是常量)。

我们可以在平面直角坐标系上点出下列点:((dp_0, ext{sumC}_0),(dp_1, ext{sumC}_1),cdots,(dp_{i-1}, ext{sumC}_{i-1})) 还有斜率 (k) 表示的直线:

Dh7nSI.png

P.S. 以下所有图的横坐标都是 ( ext{sumC}_j),纵坐标都是 (dp_j) .

我们将 (b) 改变,直线将会滑动:

D58Jkn.gif

注意最小的 (j) 正好就是滑动时第一次遇到的点。

每次更新的时候后面都会插入点,注意到斜率 (k) 和插入点的横坐标都是 递增 的,所以我们可以将相邻两点的直线连上,然后把上面的点全部去掉(因为不可能会成为最小的 (j) 了):

D5G7b4.png

发现绿线那里正好连成了一个下凸壳(凸包的定义:一个多边形是凸包当且仅当对于它的所有边满足所有点都在它所在的直线的一侧)。

现在我们怎么找遇到的第一个点呢?

注意到凸包相邻两点间的的斜率是递增的,手玩或者找规律可以得到答案就是 第一个斜率 (>k) 的点 所以我们可以二分。

当然,我们还有一种办法。

这个问题相当于在一个单调队列中找第一个大于 (k) 的点。

策略 (1):在查询的时候,可以把队头小于当前斜率的点全部删掉(因为不可能会参与答案了)
策略 (2):在插入的时候,将队尾所有不在凸包(不满足凸包性质)的点全部删掉(不要删掉插入进去的点)

不满足凸包性质的判断是纵坐标高于两点并且横坐标在两点之间,比如下图中,加入新点 (N),这使得 (G) 不满足性质,应当删去。

DImjq1.png

就按这写代码即可,注意条件:

  • 策略 (1)(dfrac{dp_2-dp_1}{C_2-C_1}le ext{sumT}_i+s)
  • 策略 (2)(dfrac{dp_{tail}-dp_{tail-1}}{C_{tail}-C_{tail-1}}gedfrac{dp_{i}-dp_{tail}}{C_{i}-C_{tail}})(其中 (tail) 是队尾)

上面是按斜率式写的,应该很容易理解,但是代码里应该注意式子有除法要移项变成乘法减少误差。

Code:

#include<ctime>
#include<queue>
#include<stack>
#include<cmath>
#include<iterator>
#include<cctype>
#include<vector>
#include<map>
#include<set>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<bitset>
using namespace std;
const int N=3e5+5;
typedef long long ll; // 开 long long
int n,s,t[N],c[N];
ll sumT[N],sumC[N],dp[N],q[N];
int main()
{
	scanf("%d%d",&n,&s);
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d",t+i,c+i);
		sumT[i]=sumT[i-1]+t[i]; sumC[i]=sumC[i-1]+c[i]; // 前缀和
	} int head=0,tail=0; // q[0]=0; 这句是隐式的,不用写
	for (int i=1;i<=n;i++)
	{
		while ((head<tail)&&(dp[q[head+1]]-dp[q[head]]<=(sumT[i]+s)*(sumC[q[head+1]]-sumC[q[head]]))) ++head; // 策略 1
		int j=q[head];
		dp[i]=dp[j]+sumT[i]*(sumC[i]-sumC[j])+s*(sumC[n]-sumC[j]); // 转移,此时的 j(也就是 q[head])已经是最小的 j 了,所以不用加 min 了
		while ((head<tail)&&((dp[q[tail]]-dp[q[tail-1]])*(sumC[i]-sumC[q[tail]])>=
		                     (dp[i]-dp[q[tail]])        *(sumC[q[tail]]-sumC[q[tail-1]]))) --tail; // 策略 2
		q[++tail]=i; // 最后再入队(因为策略 2 里不能把插入的点丢出去)
	}
	printf("%lld",dp[n]);
	return 0;
}


如果像 SDOI2012 任务安排 那样,(T_i) 是负数(时 间 倒 流),那么单调性就没了,就只能二分了 qwq

将原代码里的

while ((head<tail)&&(dp[q[head+1]]-dp[q[head]]<=(sumT[i]+s)*(sumC[q[head+1]]-sumC[q[head]]))) ++head;
int j=q[head];

换成

int l=head,r=tail;
while (l<r)
{
	int mid=(l+r)>>1;
	if (dp[q[mid+1]]-dp[q[mid]]>(sumT[i]+s)*(sumC[q[mid+1]]-sumC[q[mid]])) r=mid;
	else l=mid+1;
} int j=q[r];

即可。

那么如果 (C_i) 是负数呢?

可以倒序 dp,设计一个状态转移方程,让 ( ext{sumT}_i) 是横坐标,( ext{sumC}_i) 是斜率中的一项。仍然可以用单调队列维护凸壳,用二分法求出最优决策。

那么如果 (T_i,C_i) 都是负数呢?

这样就只能用平衡树维护了,具体可以参考 NOI2007 货币兑换。


总结一下,斜率优化是解决形如

[dp_i=min_{L(i)le jle R(i)}{dp_j+S(i,j)} ]

其中 (L(i),R(i)) 是关于 (i) 的一次函数,(S(i,j)) 是关于 (i,j) 的多项式(可以有 (i,j) 的乘积项)。

(这好像也叫 1D/1D 型转移?)

还有一个特点:斜率优化的转移方程一般很复杂 qwq


习题:Codeforces 311B Cats Transport

原文地址:https://www.cnblogs.com/CDOI-24374/p/14070590.html