【bzoj3156】 防御准备

http://www.lydsy.com/JudgeOnline/problem.php?id=3156 (题目链接)

题意

　　给出n个防御节点，每个节点有两种选择，可以花费a[i]建立一个防御塔，或者放置一个木偶，木偶的花费为到右端第一个防御塔的距离。求最少花费。

Solution

　　容易写出dp方程：$${f[i]=Min(f[j]+\frac{(i-j)*(i-j-1)}{2})}$$

　　其中${f[i]}$表示在${i}$处放置防御塔，前${i}$个节点已经完成防御所需要的最少花费。

　　易证决策单调性，划出斜率式：$${i>=\frac{(2*f[j]+j+j*j)-(2*f[k]+k+k*k)}{2*(j-k)}}$$

　　其中${j<k<i}$。

细节

　　斜率里面${j*j}$以及${k*k}$记得开long long，用一个错的程序拍了好久→_→。。

代码

// bzoj3156

#include<algorithm>

#include<iostream>

#include<cstdlib>

#include<cstring>

#include<cstdio>

#include<cmath>

#define LL long long

#define MOD 100000000

#define inf 2147483640

#define Pi acos(-1.0)

#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);

using namespace std;

const int maxn=1000010;

LL a[maxn],q[maxn],n;

LL f[maxn];

double slope(LL j,LL k) {

	return ((2.0*f[j]+j+j*j)-(2.0*f[k]+k+k*k))/(2.0*(j-k));

}

int main() {

	scanf("%lld",&n);

	for (int i=1;i<=n;i++) scanf("%lld",&a[i]);

	int l=1,r=1;q[1]=0;

	for (int i=1;i<=n;i++) {

		while (l<r && i>=slope(q[l],q[l+1])) l++;

		f[i]=f[q[l]]+(i-q[l])*(i-q[l]-1)/2+a[i];

		while (l<r && slope(q[r-1],q[r])>slope(q[r],i)) r--;

		q[++r]=i;

	}

	//for (int i=1;i<=n;i++) printf("%d : %lld\n",i,f[i]);

	printf("%lld",f[n]);

	return 0;

}