Hdu3057（划分类DP+基础斜率优化）

题目大意

将$a[1..n]$划分成若干个区间，每段区间$[i,j]$的代价为$(c[i]+c[i+1]+c[i+2]+\dots +c[j]{)}^2+m$。其中$m$为常数。
求最小的总代价和。
$a[i]>0$
$n\ge 5\ast 10^4$

---

思路

首先，这是个很明显的划分类$DP$。
直接上状态：$f[i]$表示前i个数的最小代价。
易得$f[i]=minf[j]+(s[i]-s[j]{)}^2+m$
初值：$f[0]=0$
终值：$f[n]$

---

-----------华------------丽------------丽------------的------------分------------割------------线-----------

---

然后，我们发现，这是个非常经典的$1D/1D$动态规划
显然是无法通过$5\ast 10^4$级别的数据。
考虑优化，对于一道一维状态的$DP$题，优化方向只有一个——转移。
如何减少重复或不必要的枚举呢？
设$1<j_1<j_2<i$
则对于$i$的决策，$j_2$比$j_1$优等价于满足
$f[j2]+(s[i]-s[j2]{)}^2+m<f[j1]+(s[i]-s[j1]{)}^2+m$
简单的打开并整理过后就得到了
$(f[j_2]-f[j_1])+(s[j_2{]}^2-s[j_1{]}^2)$
--------------------------------------------------- $<s[i]$
$(s[j2]-s[j1])\ast 2$
设不等式左边$=g(j_1,j_2)$
先考虑用代数
若$g(j_1,j_2)>g(j_2,j_3)$则由上面的推论可以知道对于i的决策，$j_1$比$j_2$优，$j_3$比$j_2$优，即$j_2$永远不可能成为最优的转移
又因为$a[i]>0$，即$s[i]<s[i+1]<\dots <s[n]$
则$j_2$永远不可能成为今后的最优转移
因此，我们可以通过不转移形如$g(j_1,j_2)>g(j_2,j_3)$的$j_2$
换言之，我们要转移的应该是满足$g(j_1,j_2)\le g(j_2,j_3)\le \dots$
用一个单调队列即可完成维护，每次在队首取值即可
队列中存的是决策点的下标，以队列中相邻两元素构成的$g()$为关键字单调递增
以 $g(j_1,j_2)\le g(j_2,j_3)$为例
①当$s[i]\le g(j_1,j_2)\le g(j_2,j_3)$时，$j_1$比$j_2$优，$j_2$比$j_3$优 取$j_1$
②当$g(j_1,j_2)\le s[i]\le g(j_2,j_3)$时，$j_2$比$j_1$优，$j_2$比$j_3$优 弹出$j_1$取$j_2$
③当$g(j_1,j_2)\le g(j_2,j_3)\le s[i]$时，$j_3$比$j_2$优，$j_2$比$j_1$优 弹出$j_2$取$j_3$

---

然后再考虑用较为简便的“数形结合”

发现这恰好与上面的代数一一对应，将g看成一条直线，将s[i]也看成一条直线
分别讨论当$s[i]$的斜率①$x<j_1{j}_2$②$j_1{j}_2<x<j_2{j}_3$ ③$x>j_2{j}_3$
最终可得与上文相同结论

代码

#include<cstdio>
using namespace std;
const int maxn = 500005;
long long s[maxn] , f[maxn];
int a[maxn] , q[maxn];
inline int read()
{
   char ch = getchar();
   while(ch < '0' || ch > '9') ch = getchar();
   int x = 0;
   while(ch >= '0' && ch <= '9') x = x * 10 + ch - 48 , ch = getchar();
   return x;
}
long long sqr(long long x)
{
   return x * x;
}
bool cmp1(int j2 , int j1 , int k)
{
   return (f[j2] - f[j1] + sqr(s[j2]) - sqr(s[j1])) <= k * 2 * (s[j2] - s[j1]);
}
bool cmp2(int j3 , int j2 , int j1)
{
   return (f[j3] - f[j2] + sqr(s[j3]) - sqr(s[j2])) * (s[j2] - s[j1]) <= (f[j2] - f[j1] + sqr(s[j2]) - sqr(s[j1])) * (s[j3] - s[j2]);
}
int main()
{
   int n = read() , m = read();
   s[0] = 0;
   for(int i = 1;i <= n;i++) a[i] = read() , s[i] = s[i - 1] + a[i];
   int head = 1 , tail = 1;
   q[1] = 0;
   f[0] = 0;
   for(int i = 1;i <= n;i++)
   {
   	while(head < tail && cmp1(q[head + 1] , q[head] , s[i])) head++;
   	f[i] = f[q[head]] + (s[i] - s[q[head]]) * (s[i] - s[q[head]]) + m;
   	while(head < tail && cmp2(i , q[tail] , q[tail - 1])) tail--;
   	q[++tail] = i;
   }
   printf("%lld\n",f[n]);
   return 0;
}