BZOJ3437: 小P的牧场（斜率优化DP）

3437: 小P的牧场

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1914  Solved: 1046
[Submit][Status][Discuss]

Description

小P在MC里有n个牧场，自西向东呈一字形排列（自西向东用1…n编号），于是他就烦恼了：为了控制这n个牧场，他需要在某些牧场上面建立控制站，每个牧场上只能建立一个控制站，每个控制站控制的牧场是它所在的牧场一直到它西边第一个控制站的所有牧场（它西边第一个控制站所在的牧场不被控制）（如果它西边不存在控制站，那么它控制西边所有的牧场），每个牧场被控制都需要一定的花费（毕竟在控制站到牧场间修建道路是需要资源的嘛~），而且该花费等于它到控制它的控制站之间的牧场数目（不包括自身，但包括控制站所在牧场）乘上该牧场的放养量，在第i个牧场建立控制站的花费是ai，每个牧场i的放养量是bi，理所当然，小P需要总花费最小，但是小P的智商有点不够用了，所以这个最小总花费就由你来算出啦。

Input

第一行一个整数 n 表示牧场数目

第二行包括n个整数，第i个整数表示ai

第三行包括n个整数，第i个整数表示bi

Output

只有一行，包括一个整数，表示最小花费

Sample Input

4
2424
3142

Sample Output

9
样例解释
选取牧场1，3，4建立控制站，最小费用为2+(2+1*1)+4=9。
1<=n<=1000000, 0 < a i ,bi < = 10000

HINT

Source

KpmCup#0 By Greens

没什么好说的了感觉。

#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=1000010;
int a[maxn],b[maxn],q[maxn],top,head,N; ll X[maxn],sum[maxn],dp[maxn];
ll DY(int i){ return dp[i]+X[i];}
ll getans(int i,int j){
    return dp[j]+a[i]+(sum[i]-sum[j])*i-(X[i]-X[j]);
}
int main()
{
    scanf("%d",&N);
    rep(i,1,N) scanf("%d",&a[i]);
    rep(i,1,N) scanf("%d",&b[i]);
    rep(i,1,N) {
        sum[i]=sum[i-1]+b[i];
        X[i]=X[i-1]+1ll*b[i]*i;
    }
    rep(i,1,N){
        while(head<top&&getans(i,q[head])>=getans(i,q[head+1])) head++;
        dp[i]=getans(i,q[head]);
        while(top>0&&(DY(i)-DY(q[top]))*(sum[i]-sum[q[top-1]])<(DY(i)-DY(q[top-1]))*(sum[i]-sum[q[top]])) top--;
        q[++top]=i;
    }
    printf("%lld
",dp[N]);
    return 0;
}

二分的版本，在斜率不送单调的时候用：

#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=1000010;
ll a[maxn],b[maxn],q[maxn],top,N; ll X[maxn],sum[maxn],dp[maxn];
ll DY(int i){ return dp[i]+X[i];}
ll getans(int i,int j){
    return dp[j]+a[i]+(sum[i]-sum[j])*i-(X[i]-X[j]);
}
int main()
{
    scanf("%lld",&N);
    rep(i,1,N) scanf("%lld",&a[i]);
    rep(i,1,N) scanf("%lld",&b[i]);
    rep(i,1,N) {
        sum[i]=sum[i-1]+b[i];
        X[i]=X[i-1]+b[i]*i;
    }
    rep(i,1,N){
        int L=0,R=top,Mid,pos=top;
        while(L<=R){
            Mid=(L+R)>>1;
            if(getans(i,q[Mid])<=getans(i,q[Mid+1])) pos=Mid,R=Mid-1;
            else L=Mid+1;
        }
        dp[i]=getans(i,q[pos]);
        while(top>0&&(DY(i)-DY(q[top]))*(sum[i]-sum[q[top-1]])<(DY(i)-DY(q[top-1]))*(sum[i]-sum[q[top]])) top--;
        q[++top]=i;
    }
    printf("%lld
",dp[N]);
    return 0;
}