BZOJ 1096 仓库建设 斜率DP

点击打开链接

题意:n个工厂在一条直线上,第i个工厂有p[i]件产品.坐标为x[i],在第i个工厂建立仓库花费为c[i].

现在每个工厂的产品要么到自己建立的仓库中要么到其下一个编号比它大的仓库中.(运输费用为距离*运输数量)
n<=1e6,x[i],p[i],c[i]<=1e9 问如何安排建立仓库 使得总费用最小(运输+建立仓库).




因为只能编号小的只能往编号大的运 所以可以设置dp[i] 前i个需要的最小费用 并且在第i个建立工厂.
dp[i][1]=c[i]+min(dp[j]]+cost(j+1,i-1)).
若j为i向左第一个建立工厂的,[j+1..i-1]也运输到i, 令cost[j+1][i-1]为这部分费用.
dp[i]=min(dp[j]+p[j+1]*(x[i]-x[j+1])+...p[i-1]*(x[i]-x[i-1]))




dp[i]=min(dp[j]+x[i]*(p[j+1]+..p[i-1])-(p[j+1]*x[j+1]+..p[i-1]*x[i-1]).
用两个前缀和来维护 dp[i]=min(dp[j]+x[i]*(pre[i]-pre[j])-(sum[i]-sum[j])) j<i. 
O(n^2) Tle 尝试用斜率来优化.



dp[i]=pre[i]+sum[i]+min(dp[j]+sum[j]-pre[j]*x[i])
则令M=dp[j]+sum[j]-pre[j]*x[i].
dp[j]-sum[j]=x[i]*pre[j]+M.
x[i]看作斜率 决策j看作点(pre[j],dp[j]-sum[j]) 
我们要求最小的截距,也就是将斜率为x[i]的直线 从下(负无穷)开始移动 碰到的第一个点(决策),其对应的截距为最小截距.
则所有能构成决策的点,理应构成一个下凸壳,则当slope(q[r-1],q[r])>slope(q[r],i])时 弹出q[r].

在挑选最优决策点时，由于斜率x[i]是递增的，所以每次的最优决策点前的所有决策点都是没用的，可以利用这个性质维护一个队列，队头就是最优决策点。(mark.)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> ii;
const int N=1e6+20,inf=0x3f3f3f3f;
ll x[N],c[N],p[N],dp[N],pre[N],sum[N],q[N];
double slope(int j,int k)
{
	return double(dp[j]+sum[j]-dp[k]-sum[k])/double(pre[j]-pre[k]);
}
int main()
{
	int n;
	cin>>n;
	for(int i=1;i<=n;i++)
		scanf("%I64d%I64d%I64d",&x[i],&p[i],&c[i]);
	for(int i=1;i<=n;i++)
		pre[i]=pre[i-1]+p[i];
	for(int i=1;i<=n;i++)
		sum[i]=p[i]*x[i]+sum[i-1];
	
	int l=0,r=0;
	//dp[0]=0,q[0]=0;
	for(int i=1;i<=n;i++)
	{
		while(l<r&&slope(q[l+1],q[l])<x[i])
			l++;
		int t=q[l];
		dp[i]=dp[t]-sum[i]+sum[t]+x[i]*(pre[i]-pre[t])+c[i];
		while(l<r && slope(q[r-1],q[r])>slope(q[r],i))
			r--;
		q[++r]=i;
	}
	cout<<dp[n]<<endl;	
	
	return 0;
}