斜率优化dp 任务安排 luogu2365 和5785

WA＝=Wonderful Answer

已于 2023-05-14 17:33:45 修改

阅读量78

点赞数

文章标签：算法 c++ 开发语言

于 2023-05-14 16:48:14 首次发布

本文链接：https://blog.csdn.net/m0_64675532/article/details/130670575

版权

文章介绍了一种利用斜率优化处理动态规划问题的方法，特别是在数据范围较大的情况下。通过维护一个斜率递增的队列，文章详细阐述了如何通过二分查找找到合适的截距，从而降低时间复杂度到O(NlogN)。代码示例展示了如何在C++中实现这一算法。

摘要由CSDN通过智能技术生成

这是斜率优化模板题，不解释。

#include <bits/stdc++.h>
typedef long long ll;

using namespace std;
#define pb push_back
#define inf 0x3f3f3f3f
#define endl "\n"
#define d(a) cout<<#a<<a<<endl;
#define max(a, b) a > b ? a : b
int n,s,q[5010];
ll ti[5010],c[5010],f[5010];

double slope(int i, int j){
	return 1.0*(f[i]-f[j])/(c[i]==c[j]?1e-9:c[i]-c[j]);
	
	
}


int main(){
	cin>>n>>s;
	for (int i=1;i<=n;i++){
		cin>>ti[i]>>c[i];
		ti[i]+=ti[i-1],c[i]+=c[i-1];
	}
	
	int h=1,t=0;
	for (int i=1;i<=n;i++){
		while (h<t and slope(i-1,q[t])<=slope(q[t],q[t-1])) t--;
		q[++t]=i-1;//下标入队，便于队头出队
		while (h<t and slope(q[h+1],q[h])<=ti[i]+s) h++;
		int j=q[h];//队头出队
		f[i]=f[j]+ti[i]*(c[i]-c[j])+s*(c[n]-c[j]);
		
	}
	
	cout<<f[n]<<endl;
}

当数据范围是这样的，5785：

递推公式还是原本的公式，对于前i个任务完成的最短时间dp[i]

dp[i]=dp[j]+ti[i]*(c[i]-c[j])+s*(c[n]-c[j])

改写成dp[j]=(ti[i]+s)*c[j]+dp[i]-ti[i]*c[i]-s*c[n]

其中，ti[i]是完成前i个任务的总时间，c[i]是完成前i个任务的总时间系数，这俩都是前缀和。

注意到Ti是有可能为负数的，这个时候斜率ti[i]+s就不是递增的了，我们寻找最小截距的时候，需要二分的去查找第一个大于当前斜率的点，能二分的原因是因为我们维护的队列斜率是递增的，时间复杂度O（NlogN）。

代码：

#include <bits/stdc++.h>
typedef long long ll;

using namespace std;
#define pb push_back
#define inf 0x3f3f3f3f
#define endl "\n"
#define d(a) cout<<#a<<a<<endl;
#define max(a, b) a > b ? a : b
const int maxn=3e5+10;
int n,s,q[maxn],h=1,t=0;
ll ti[maxn],c[maxn],f[maxn];



ll dy(int x,int y) {
	return f[y]-f[x];
} 

ll dx(int x,int y){
	return c[y]-c[x];
}

int find1(int i){
	if (h==t) return q[h];
	int l=h-1,r=t+1;
	while(l+1<r){
		int mid=r+l>>1;
		if (dy(q[mid-1],q[mid])<=dx(q[mid-1],q[mid])*(ti[i]+s)) l=mid;
		else r=mid;
	}
	return q[l];
}

int main(){
	cin>>n>>s;
	for (int i=1;i<=n;i++){
		cin>>ti[i]>>c[i];
		ti[i]+=ti[i-1];
		c[i]+=c[i-1];
		
	}
	
	for ( int i=1;i<=n;i++){
		
	while (h<t and (f[i-1]-f[q[t]])*dx(q[t-1],q[t])<=(c[i-1]-c[q[t]])*dy(q[t-1],q[t])) t--;
	q[++t]=i-1;
	int j=find1(i);
	f[i]=f[j]+ti[i]*(c[i]-c[j])+s*(c[n]-c[j]);
	}cout<<f[n]<<endl;
	
}