[斜率优化DP]bzoj1911 - cool's Blog

[斜率优化DP]bzoj1911

cool posted @ 2016年3月12日 13:37 in DP , 392 阅读

1911: [Apio2010]特别行动队

Time Limit: 4 Sec  Memory Limit: 64 MB
Submit: 3390  Solved: 1540
[Submit][Status][Discuss]

Description

Input

Output

Sample Input

4
-1 10 -20
2 2 3 4

Sample Output

9

HINT

 

这是一道DP题,还是比较明显的= =

f[i]=max(f[j]+a*(sum[i]-sum[j])^2+b(sum[i]-sum[j])+c);

然后呢 一看这样的式子我们就可以尝试斜率优化。

设k<j<i

若f[j]>f[k],则k无用

(f[k]+a*s[k]^2-b*s[k]-f[j]-a*s[j]^2+b*s[k])/2*a*(s[k]-s[j]) <= s[i]

显然就可以斜率优化了

#include <cstdio>
#define maxn 1000010
#define ll long long
ll f[maxn],s[maxn],q[maxn],a,b,c;
ll dy(int x)
{
	return f[x]+a*s[x]*s[x]-b*s[x];
}
ll dx(int x)
{
	return 2*a*s[x];
}

int main()
{
	int n;
	scanf("%d",&n);
	scanf("%lld%lld%lld",&a,&b,&c);
	for (int i = 1;i <= n;i++)
	{
		ll x;
		scanf("%lld",&x);
		s[i] = s[i-1]+x;
	}
	int l = 1,r = 1;
	for (int i = 1;i <= n;i++)
	{
		for (;l<r&&(dy(q[l+1])-dy(q[l]))>=s[i]*(dx(q[l+1])-dx(q[l]));l++);
		f[i] = f[q[l]]+(s[i]-s[q[l]])*(s[i]-s[q[l]])*a+(s[i]-s[q[l]])*b+c;
		for (;l<r&&(dy(i)-dy(q[r]))*(dx(q[r])-dx(q[r-1]))<=(dy(q[r])-dy(q[r-1]))*(dx(i)-dx(q[r]));r--);
		q[++r] = i;
	}
	printf("%lld\n",f[n]);
	return 0;
}

dp提高还是得多艹题!


登录 *


loading captcha image...
(输入验证码)
or Ctrl+Enter
Host by is-Programmer.com | Power by Chito 1.3.3 beta | © 2007 LinuxGem | Design by Matthew "Agent Spork" McGee