【BZOJ3437】小P的牧场 斜率优化

【BZOJ3437】小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
2 4 2 4
3 1 4 1

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]) (1≤j<i,j<k<i)

然后我们将Σ拆开，变成i*Σb[k]+Σb[k]*k，这个可以用前缀和维护

设sb是b[k]的前缀和，sk是b[k]*k的前缀和，再得到方程

f[i]=f[j]+i*(sb[i-1]-sb[j])-sk[i-1]+sk[j]+a[i]

在整理一下就行了

#include <cstdio>
#include <iostream>
#include <cstring>
#define y(_) (f[_]+sk[_])
#define x(_) sb[_]
using namespace std;
const int maxn=1000010;
int n,h,t;
long long a[maxn],b[maxn],sb[maxn],sk[maxn],q[maxn],f[maxn];
int main()
{
	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]);
		sb[i]=sb[i-1]+b[i];
		sk[i]=sk[i-1]+b[i]*i;
	}
	h=t=1;
	for(i=1;i<=n;i++)
	{
		while(h<t&&y(q[h+1])-y(q[h])<=i*(x(q[h+1])-x(q[h])))	h++;
		f[i]=f[q[h]]+i*(sb[i-1]-sb[q[h]])-sk[i-1]+sk[q[h]]+a[i];
		while(h<t&&(y(q[t])-y(q[t-1]))*(x(i)-x(q[t]))>=(y(i)-y(q[t]))*(x(q[t])-x(q[t-1])))	t--;
		q[++t]=i;
	}
	printf("%lld",f[n]);
	return 0;
}