HDU3507(斜率优化dp 入门)

解题思路:第一次做斜率优化dp,参考了博客http://www.cnblogs.com/kuangbin/archive/2012/08/26/2657650.html。使原本O(n^2)的复杂度变成O(n)。看完博客之后,看了一下代码发现代码中(yj-yk)/(xj-xk) <= sum[i],只要碰到一个不符合这个条件就用j来算dp[i]了。其实是这样的,因为符合上面那个不等式,代表j在以后都会比k要优,所以可以把k去掉,但是一旦在j的时候不满足这个等式,那么维护时斜率本来就是递增的,这意味着后面的点j'和k的斜率只会越来越大,更加满足不了要求,还是k是最优的。这意味着k比它前面的要优,比它后面的点也要优。顾就是答案。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 500009
using namespace std;
int sum[N],stk[N],dp[N];
int head,tail,m;
int getdp(int i,int j)
{
	return dp[j]+(sum[i]-sum[j])*(sum[i]-sum[j])+m;
}
int getup(int j,int i)
{
	return dp[j]+sum[j]*sum[j]-dp[i]-sum[i]*sum[i];
}
int getdown(int j,int i)
{
	return 2*(sum[j]-sum[i]);
}
int main()
{
	//freopen("t.txt","r",stdin);
	int n;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		sum[0]=0;
		for(int i=1;i<=n;i++) scanf("%d",&sum[i]);
		for(int i=1;i<=n;i++)
		{
			sum[i]+=sum[i-1];
		}
		head=tail=0;
		stk[tail++]=0;
		for(int i=1;i<=n;i++)
		{
			while(head+1<tail&&getup(stk[head+1],stk[head])<=sum[i]*getdown(stk[head+1],stk[head]))//除法会涉及浮点运算,拖慢速度,所以用乘法 
			head++;
			dp[i]=getdp(i,stk[head]);
			while(head+1<tail&&getup(i,stk[tail-1])*getdown(stk[tail-1],stk[tail-2])<=getup(stk[tail-1],stk[tail-2])*getdown(i,stk[tail-1]))
			tail--;
			stk[tail++]=i;
		}
		printf("%d\n",dp[n]);
	}
	return 0;
} 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值