3156: 防御准备

自从有了斜率优化,做DP题,手也不冷了,腿也不痛了,腰也不酸了,分分钟搞定啊。

公式随便推一下就好了,倒着DP,唯一坑爹的是f要开两维,分别表示当前点放或不放塔,不放塔只能从前面(后面?)放塔的状态转移过来。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=1000010;
const int inf=1e9;
inline int read(){
	int x=0;char ch;
	while(ch<'0'||ch>'9')ch=getchar();
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x;
}
typedef long long ll;
ll f[N][2];													//0 木偶 1塔 
int q[N],head,tail,n,A[N];
double slop(int j,int k){
	return (double)(f[j][1]-f[k][1])/(j-k)+(double)(k+j+1)/2;
}
int main(){
	n=read();
	for(int i=1;i<=n;i++)A[i]=read();
	head=1;tail=1;
	q[head]=n;f[n][1]=A[n];f[n][0]=inf;
	for(int i=n-1;i>=1;i--){
		while(head<tail&&slop(q[head+1],q[head])>i)head++;
		int t=q[head];
		f[i][0]=f[t][1]+(ll)(t-i)*(t-i+1)/2;
		f[i][1]=min(f[i+1][1],f[i+1][0])+A[i];
		while(head<tail&&slop(q[tail-1],q[tail])<slop(q[tail],i))tail--;
		q[++tail]=i;
	}
	printf("%lld",min(f[1][0],f[1][1]));
	return 0;
}


你可能感兴趣的:(3156: 防御准备)