BZOJ1096 [ZJOI2007] 仓库建设

【BZOJ1096】[ZJOI2007]仓库建设

Description

L公司有N个工厂，由高到底分布在一座山上。如图所示，工厂1在山顶，工厂N在山脚。由于这座山处于高原内陆地区（干燥少雨），L公司一般把产品直接堆放在露天，以节省费用。突然有一天，L公司的总裁L先生接到气象部门的电话，被告知三天之后将有一场暴雨，于是L先生决定紧急在某些工厂建立一些仓库以免产品被淋坏。由于地形的不同，在不同工厂建立仓库的费用可能是不同的。第i个工厂目前已有成品Pi件，在第i个工厂位置建立仓库的费用是Ci。对于没有建立仓库的工厂，其产品应被运往其他的仓库进行储藏，而由于L公司产品的对外销售处设置在山脚的工厂N，故产品只能往山下运（即只能运往编号更大的工厂的仓库），当然运送产品也是需要费用的，假设一件产品运送1个单位距离的费用是1。假设建立的仓库容量都都是足够大的，可以容下所有的产品。你将得到以下数据：1：工厂i距离工厂1的距离Xi（其中X1=0）;2：工厂i目前已有成品数量Pi;:3：在工厂i建立仓库的费用Ci;请你帮助L公司寻找一个仓库建设的方案，使得总的费用（建造费用+运输费用）最小。

Input

第一行包含一个整数N，表示工厂的个数。接下来N行每行包含两个整数Xi, Pi, Ci, 意义如题中所述。

Output

仅包含一个整数，为可以找到最优方案的费用。

Sample Input

3 0 5 10 5 3 100 9 6 10

Sample Output

32

HINT

在工厂1和工厂3建立仓库，建立费用为10+10=20，运输费用为(9-5)*3 = 12，总费用32。如果仅在工> 厂3建立仓库，建立费用为10，运输费用为(9-0)*5+(9-5)*3=57，总费用67，不如前者优。
【数据规模】 对于100%的数据， N ≤1000000。 所有的Xi, Pi,Ci均在32位带符号整数以内，保证中间计算结果不超过64位带符号整数。

Solution：

dp方程：$dp[i]=dp[j]+w[j+1,i]+C[i]$
w[i,j]可以用前缀和O(n)处理：

• $sum[i]=sum[i+1]+(X[n]-X[i])*P[i]$
• $K[i]=K[i-1]+P[i]$

斜率公式：

$$g(k,j)=\frac{dp[j]-dp[k]+sum[j+1]-sum[k+1]}{K[k]-K[j]}>=(X[n]-X[i])(j<k)$$
斜率优化不再阐述。

#include<stdio.h>
#include<string.h>
#include<iostream>
#define ll long long
#define M 1000005
using namespace std;
int X[M],P[M],C[M],Q[M];
ll dp[M],sum[M],K[M];
double g(int k,int j){
    return 1.0*(dp[j]-dp[k]+sum[j+1]-sum[k+1])/(K[k]-K[j]);
}
void check(ll &a,ll b){
    if(a==-1||a>b)a=b;
}
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d %d %d",&X[i],&P[i],&C[i]);
        K[i]=K[i-1]+P[i];
    }
    for(int i=n;i>=1;i--)
        sum[i]=sum[i+1]+1LL*(X[n]-X[i])*P[i];
    memset(dp,-1,sizeof(dp));
    dp[0]=0;
    int L=0,R=-1;
    Q[++R]=0;
    for(int i=1;i<=n;i++){
        while(L<R&&g(Q[L+1],Q[L])>=X[n]-X[i])L++;
        int id=Q[L];
        dp[i]=dp[id]+C[i]+sum[id+1]-sum[i+1]-1LL*(K[i]-K[id])*(X[n]-X[i]);
        while(L<R&&g(i,Q[R])>=g(Q[R],Q[R-1]))R--;
        Q[++R]=i;
    }
    cout<<dp[n]<<endl;
    return 0;
}