斜率优化DP。
参考资料:2004周源《浅谈数形结合思想在信息学竞赛中的应用》。斜率优化部分的证明,构造下凸曲线。
本题题解:
C[i] 表示前 i 个玩具的长度
F[i] 表示前 i 个玩具装完之后的最小费用
状态转移方程: F[i] = min { F[j] + (C[i] - C[j] + i - j - 1 - L)^2 }
令 S[i] = C[i] + i 则
F[i] = min { F[j] + (S[i] - S[j] - L - 1)^2 }
令 m = S[i] - L - 1 则
F[i] = min { F[j] + (m - S[j])^2 }
求斜率方程:
设 i 之前的两个决策点位 j , k 且 j 比 k 决策更优,则:
Fj + (m - Sj)^2 <= Fk + (m - Sk)^2
转化:Fj - Fk + Sj^2 - Sk^2 <= 2*m* (Sj - Sk)
令Y = F + S^2 , X = S 则上式为:
Yj - Yk <= 2*m*(Xj - Xk)
Si 是单调递增的,所以可以将右侧除过来:
(Yj - Yk)/(Xj - Xk) <= 2*m 记为:X(j, k)<= 2*m (斜率公式)
规定队列的维护规则(构造下凸曲线)
队首维护:
队尾维护:
a.若X(B,A)<=F[I],且X(C,B)<=F[I],则C比B好,B比A好
b.若X(B,A)<=F[I],且X(C,B)>F[I],则B比C好,B比A好,B为极大值
c.若X(B,A)>F[I],A比B好
a,c情况直接删掉B,b情况保留.b情况可改为X(B,A)<X(C,B)
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long LL;
#define N 50003
#define sqr(a) (a)*(a)
LL c[N], s[N], f[N];
int n, l, q[N];
double slope(int i, int j) {
return 1.0*(f[i]-f[j]+sqr(s[i])-sqr(s[j]))/(s[i]-s[j]);
}
int main() {
while (scanf("%d%d", &n, &l) == 2) {
c[0] = s[0] = 0;
for (int i=1; i<=n; i++) {
scanf("%lld", &c[i]);
c[i] += c[i-1];
s[i] = c[i] + i;
}
int h, t;
f[0] = q[0] = h = t = 0;
for (int i=1; i<=n; i++) {
LL m = s[i] - l - 1;
while (h<t && slope(q[h+1], q[h]) <= 2*m) h++;
int j = q[h];
f[i] = f[j] + sqr(m-s[j]);
while (h<t && slope(q[t], q[t-1]) >= slope(i, q[t])) t--;
q[++t] = i;
}
printf("%lld\n", f[n]);
}
return 0;
}