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

3437: 小P的牧场

Time Limit: 10 Sec   Memory Limit: 128 MB

Submit: 870   Solved: 494

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

Sample Output

9
样例解释
选取牧场1，3，4建立控制站，最小费用为2+(2+1*1)+4=9。
数据范围与约定
对于100%的数据，1<=n<=1000000,0<ai,bi<=10000

HINT

[Submit][Status][Discuss]

题解：

斜率优化无脑题：

先列出DP式f[i]=f[j]+a[i]+[i-(j+1)]*b[j+1]+[i-(j+2)]*b[j+2]+[i-(j+3)]*b[j+3]+...+[i-(i-1)]*b[i-1]

f[i]=f[j]+a[i]+i*(b[j+1]+b[j+2]+...+b[i-1])-[(j+1)*b[j+1]+(j+2)*b[j+2]+...+(i-1)*b[i-1]]

我们维护一个b[i]的前缀和bs[i]，维护一个b[i]*i的前缀和bb[i]

原式就化成了f[i]=f[j]+i*(bs[i-1]-bs[j])-(bb[i-1]-bb[j])

对于两个元素：j,k且j>k

f[i]=f[j]+i*(bs[i-1]-bs[j])-(bb[i-1]-bb[j])，f[i]=f[k]+i*(bs[i-1]-bs[k])-(bb[i-1]-bb[k])

若决策点j比决策点k更优：f[j]+i*(bs[i-1]-bs[j])-(bb[i-1]-bb[j])<f[k]+i*(bs[i-1]-bs[k])-(bb[i-1]-bb[k])

经过整理：f[j]-f[k]-bb[k]+bb[j]<i*(bs[j]-bs[k])

可以设g[i]=f[i]+bb[i]

原式转化为：(g[j]-g[k])/(bs[j]-bs[k])<i

斜率优化DP即可

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define LL long long 
const int N=1000005;
LL xj(LL xa,LL ya,LL xb,LL yb)
{return xa*yb-xb*ya;}
LL n,a[N],bs[N],bb[N],x,Q[N],f[N],g[N],h=1,t=0;
int main()
{
	scanf("%lld",&n);
	for(LL i=1;i<=n;i++) scanf("%lld",&a[i]);
	for(LL i=1;i<=n;i++)
	{
		scanf("%lld",&x);
		bs[i]=bs[i-1]+x;
		bb[i]=bb[i-1]+i*x;
	}
	memset(f,127,sizeof(f));
	f[0]=0;
	Q[++t]=0;
	for(LL i=1;i<=n;i++)
	{
		while(t-h+1>=2&&(g[Q[h+1]]-g[Q[h]])*1.0/(bs[Q[h+1]]-bs[Q[h]])<i) h++;
		f[i]=f[Q[h]]+a[i]+i*(bs[i-1]-bs[Q[h]])-(bb[i-1]-bb[Q[h]]);
		g[i]=f[i]+bb[i];
		while(t-h+1>=2&&xj(bs[Q[t]]-bs[Q[t-1]],g[Q[t]]-g[Q[t-1]],bs[i]-bs[Q[t-1]],g[i]-g[Q[t-1]])<0) t--;
		Q[++t]=i;
	}
	printf("%lld",f[n]);
}