BZOJ 3437: 小P的牧场 斜率优化DP

Posted 2020-07-28 Meek

3437: 小P的牧场

Description

 背景

    小P是个特么喜欢玩MC的孩纸。。。

 描述

    小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。

数据范围与约定


对于100%的数据，1<=n<=1000000,0<ai,bi<=10000

题解：

　　斜率优化、

　　设定dp[i]为前在i处放控制台的答案

　　那么

　　　　dp[i] = min{dp[j]+(s[i]-s[j])*i-(c[i]-c[j])} +a[i] j<i

　　其中 s[i] 为b[i]的前缀和,c[i]为b[i]*i的前缀和

　　单调性就不证明了,和BZOJ3156特像

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 1e6+20,inf = 2e9, mod = 1e9+7;
typedef long long ll;

ll dp[N],s[N],c[N],a[N],b[N],n;
int l,r,q[N];
double getsum(int j,int k)
{
    return (double) (dp[j]-dp[k]+c[j]-c[k]) / (double)(s[j]-s[k]);
}
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=1;i<=n/2+n%2;i++) swap(a[i],a[n-i+1]),swap(b[i],b[n-i+1]);
    for(int i=1;i<=n;i++) s[i]=s[i-1]+b[i];
    for(int i=1;i<=n;i++) c[i]=c[i-1]+b[i]*i;

    for(int i=1;i<=n;i++)
    {
        while(l<r&&getsum(q[l],q[l+1])<i) l++;
        int now=q[l];
        dp[i]=dp[now]+(s[i]-s[now])*i-(c[i]-c[now])+a[i];
        while(l<r&&getsum(i,q[r])<getsum(q[r],q[r-1])) r--;
        q[++r]=i;
    }
    printf("%lld\n",dp[n]);
    return 0;
}