DP的斜率优化

最新推荐文章于 2024-02-06 11:44:45 发布

Freopen

最新推荐文章于 2024-02-06 11:44:45 发布

阅读量122

点赞数

分类专栏： DP DP优化模板

本文链接：https://blog.csdn.net/qq_35950004/article/details/83717522

版权

模板同时被 3 个专栏收录

212 篇文章 0 订阅

订阅专栏

105 篇文章 0 订阅

订阅专栏

DP优化

47 篇文章 0 订阅

订阅专栏

HDU - 3507

#include<cstdio>
#include<cstring>
#include<cctype>
#include<algorithm>
#define maxn 1000005
using namespace std;

int dp[maxn];
int sta[maxn];

int h=0,t=0;
int n,m;
long long sum[maxn];

double g(int a){
	return dp[a]+sum[a]*sum[a];
}

int main(){
	int u;
	while(~scanf("%d%d",&n,&m)){
		h=t=0;
		sta[t++]=0;
		for(int i=1;i<=n;i++){
			scanf("%d",&u);
			sum[i]=sum[i-1]+u;
			while(h<t-1 && (g(sta[h+1])-g(sta[h]))<=1ll*2*sum[i]*(sum[sta[h+1]]-sum[sta[h]]))
				h++;
			dp[i]=dp[sta[h]]+(sum[i]-sum[sta[h]])*(sum[i]-sum[sta[h]])+m;
			while(h<t-1 && (g(sta[t-1])-g(sta[t-2]))*(sum[i]-sum[sta[t-1]])>=(g(i)-g(sta[t-1]))*(sum[sta[t-1]]-sum[sta[t-2]]))
				t--;
			sta[t++]=i;
		}
		printf("%d\n",dp[n]);
	}
}