BZOJ3437:小P的牧场(斜率优化DP)

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
2 4 2 4
3 1 4 2

Sample Output

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

Source

KpmCup#0 By Greens

Solution

1A就很开心了

非常裸的一眼斜率优化DP

设$sum_i$为$b_i$的后缀和，$s_i$为$(n-i)*b_i$的后缀和

那么DP式子很明显可以写为

$f_i=f_j+s_{j+1}-s_i-(n-i)*(sum_{j+1}-sum_{i+1})+a_i$

感性理解一下其实式很子好懂

把式子拆开后发现满足斜率单调性，直接上斜率优化就好了

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#define N (1000000+1000)
#define LL long long
using namespace std;

LL n,a[N],b[N],sum[N],s[N],q[N],f[N]; 

LL K(LL j){return sum[j+1];}
LL B(LL j){return f[j]+s[j+1]-n*sum[j+1];}
LL Y(LL i,LL j){return K(j)*i+B(j);}

bool cover(LL x1,LL x2,LL x3)
{
    LL w1=(B(x2)-B(x1))*(K(x1)-K(x3));
    LL w2=(B(x3)-B(x1))*(K(x1)-K(x2));
    return w1<=w2;
}

int main()
{
    scanf("%lld",&n);
    for (int i=1; i<=n; ++i) scanf("%lld",&a[i]);
    for (int i=1; i<=n; ++i) scanf("%lld",&b[i]);
    for (int i=n; i>=1; --i)
    {
        sum[i]=sum[i+1]+b[i];
        s[i]=s[i+1]+(n-i)*b[i];
    }
    int head=0,tail=0;
    for (int i=1; i<=n; ++i)
    {
        while (head<tail && Y(i,q[head])>=Y(i,q[head+1])) head++;
        f[i]=Y(i,q[head])-s[i+1]+n*sum[i+1]-i*sum[i+1]+a[i];
        while (head<tail && cover(i,q[tail],q[tail-1])) tail--;
        q[++tail]=i;
    }
    printf("%lld",f[n]);
}