【HDU3507】Print Article-DP斜率优化入门

测试地址:Print Article

题目大意:要打印一份有N(N≤500000)个词的文章,每个词有一个花费Ci,连续打一段词的花费为(ΣCi)^2+M,M为常数,求打印这份文章的最小花费。

做法:这道题要用到DP的斜率优化。

首先分析题目发现这道题显然可以用动态规划解决,设f[i]为打印前i个词的最小花费,sum[i]为前i个词的花费之和,则状态转移方程为:f[i]=min{f[j]+(sum[i]-sum[j])^2}(0≤j<i)+M,这是一个O(N^2)的式子,然而N可达500000,华丽爆炸,所以需要想办法优化。

我们把min里面的式子展开为f[j]+sum[i]^2-2*sum[i]*sum[j]+sum[j]^2,其中sum[i]^2与j无关,把它从函数里提出来,那么设剩下的式子为G,再令k=2*sum[i],x=sum[j],y=f[j]+sum[j]^2,则得G=-kx+y,所以y=kx+G,这个式子很像直线方程的斜截式,那么当G从小变大时,就相当于一条斜率为k的直线从下到上运动,每当接触到一个点(x,y)时都会产生一个合法的G,那么我们要求G的最小值,就要想办法求出这条直线运动时碰到的第一个点。

我们令x[i]=sum[i],y[i]=f[i]*sum[i]^2,(x[i],y[i])就代表着i这个状态点的坐标,可以看到k和x[i]都是单调不降的,那么我们就对于前面的所有状态点维护一个下凸壳(类似Graham-Scan做凸包),每次求直线与凸壳的切点即可。由于k单调不降,那么我们可以直观的感受到,如果一个点不是i的最优决策点,那么也不会是后面的最优决策点,就可以把i丢掉,对于这个去头去尾和在尾部添加的操作,可以用单调队列实现。由于一个点最多入队一次,出队一次,所以复杂度降为O(N),问题解决。

注意:比较直线斜率时最好不要直接求斜率来比较,用叉积比较保险,然而也不要搞混不等号的方向。另外删除无用决策点和上凸点的时候,判断叉积正负性是需要加等号的,我也不知道为什么不加就WA......

以下是本人代码:

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int n,h,t,q[500010];
long long m,sum[500010],f[500010];
struct point
{
  long long x,y;
  point operator - (point a) const
  {
    point s;
	s.x=x-a.x;
	s.y=y-a.y;
    return s;
  }
}p[500010];

long long multi(point a,point b)
{
  return a.x*b.y-b.x*a.y;
}

void solve()
{
  h=1,t=1;
  q[1]=0;p[0].x=p[0].y=0;f[0]=0;
  for(int i=1;i<=n;i++)
  {
    point now;
	now.x=1,now.y=2*sum[i];
    while(h<t&&multi(now,p[q[h+1]]-p[q[h]])<=0) h++;
	int j=q[h];
	f[i]=f[j]+(sum[i]-sum[j])*(sum[i]-sum[j])+m;
	p[i].x=sum[i],p[i].y=f[i]+sum[i]*sum[i];
	while(h<t&&multi(p[i]-p[q[t-1]],p[q[t]]-p[q[t-1]])>=0) t--;
	q[++t]=i;
  }
}

int main()
{
  while(scanf("%d%lld",&n,&m)!=EOF)
  {
    sum[0]=0;
    for(int i=1;i<=n;i++)
    {
      long long a;
	  scanf("%lld",&a);
	  sum[i]=sum[i-1]+a;
    }
  
    solve();
	
	printf("%lld\n",f[n]);
  }
  
  return 0;
}


posted @ 2017-04-29 11:44  Maxwei_wzj  阅读(90)  评论(0编辑  收藏  举报