bzoj 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

这题稍微有点麻烦

首先还是可以列出一般的转移方程：f[i]=min(f[j]+Σb[k]*(i-k))+a[i],k为j到i

我们观察这个Σ，可以把它拆分成i*Σb[k]和Σb[k]*k，这两个都是好维护的

我们用s[k]表示b[k]的前缀，ks表示b[k]*k的前缀

那么转移方程就变成了f[i]=min(f[j]+i*s[i]-i*s[j]-ks[i]+ks[j]))+a[i]

考虑当j>k时j比k更优

则有f[j]+i*s[i]-i*s[j]-ks[i]+ks[j])<f[k]+i*s[i]-i*s[k]-ks[i]+ks[k])

移项后可得((f[j]+ks[j])-(f[k]+ks[k]))/(s[j]-s[k])<i

维护即可

#include<cstdio>
#include<algorithm>
using namespace std;
long long a[1000001],b[1000001];
long long s[1000001],ks[1000001];
long long f[1000001];
int q[1000001];
inline double solve(int k,int j)
{
     return double((f[j]+ks[j])-(f[k]+ks[k]))/double(s[j]-s[k]);
}
int main()
{
     int n;
     scanf("%d",&n);
     int i;
     for(i=1;i<=n;i++)
          scanf("%lld",&a[i]);
     for(i=1;i<=n;i++)
          scanf("%lld",&b[i]);
     for(i=1;i<=n;i++)
          s[i]=s[i-1]+b[i];
     for(i=1;i<=n;i++)
          ks[i]=ks[i-1]+b[i]*i;
     int l=1,r=0;
     r++;
     q[r]=0;
     for(i=1;i<=n;i++)
     {
          while(l<r&&solve(q[l],q[l+1])<i)
               l++;
          f[i]=f[q[l]]+i*s[i]-i*s[q[l]]+ks[q[l]]-ks[i]+a[i];
          while(l<=r&&solve(q[r-1],q[r])>solve(q[r],i))
               r--;
          r++;
          q[r]=i;
     }
     printf("%lld\n",f[n]);
     return 0;
}