斜率优化动态规划可以用来解决这道题。同时这也是一道经典的斜率优化基础题。
分析:明显是动态规划。令 dp[i] d p [ i ] 为前 i i 个装箱的最小花费。
转移方程如下:
用 sum[i] s u m [ i ] 表示前 i i 个容器的长度之和(即 C C 的前缀和),方程简化为:
又令 f[i] f [ i ] 为 sum[i]+i s u m [ i ] + i ,继续简化方程为:
暴力dp是 O(n2) O ( n 2 ) ,考虑优化。如何优化,就是用前面所提到的斜率优化。这玩意到底是什么?我们先来继续对状态转移方程进行进一步的推导。
对于每个 dp[i] d p [ i ] 可以知道都是由一个 j0 j 0 推过来的。这个 j0 j 0 对于当前的 i i 是最优的决策。假设现在有两个决策 j1,j2(1≤j1<j2<i) j 1 , j 2 ( 1 ≤ j 1 < j 2 < i ) ,且决策 j2 j 2 优于 j1 j 1 ,则有:
拆开可得:
化简可得:
即:
也就是说,若 j1,j2 j 1 , j 2 满足上面这个式子,那么 j2 j 2 一定比 j1 j 1 优。
为什么叫斜率优化?因为上面这个式子可以把看作 dp[i]+g[i] d p [ i ] + g [ i ] 看做纵坐标, f[i] f [ i ] 看做横坐标,上面的等式右侧就相当于 ΔyΔx=k Δ y Δ x = k 也就是一个一次函数的斜率。当这个斜率 k≤2f[i] k ≤ 2 f [ i ] 则 j2 j 2 优于 j1 j 1 。
假如我们有三个决策 j1,j2,j3 j 1 , j 2 , j 3 (如下图)
容易证明: j2 j 2 不可能是最优的。
这样一来,每两个决策间的斜率便是单调上升的。
所以有两种做法:
注意事项:
long long.
还有 g[0]=(L+1)2 g [ 0 ] = ( L + 1 ) 2 而不是 0 0 !!!!(因为这个我调了很久)
给一下单调队列做法的代码:
#include
#include
#include
#include
using namespace std;
#define int long long
const int MAXN = 50050;
int N, L, dp[MAXN], sum[MAXN], f[MAXN], g[MAXN], h, t, Q[MAXN];
inline double xie(int j1, int j2)
{
return (double) (dp[j2] + g[j2] - dp[j1] - g[j1]) / (f[j2] - f[j1]);
}
#undef int
int main()
{
scanf("%d%d", &N, &L);
for(int i = 1; i <= N; i++)
{
scanf("%d", &sum[i]);
sum[i] += sum[i - 1];
f[i] = sum[i] + i;
g[i] = (f[i] + L + 1) * (f[i] + L + 1);
}
g[0] = (L + 1) * (L + 1); //important!!!
for(int i = 1; i <= N; i++)
{
while(h < t && xie(Q[h], Q[h + 1]) <= 2 * f[i]) h++;
dp[i] = dp[Q[h]] + (f[i] - f[Q[h]] - L - 1) * (f[i] - f[Q[h]] - L - 1); //更新dp值
while(h < t && xie(Q[t], i) < xie(Q[t - 1], Q[t])) t--;
Q[++t] = i;
}
printf("%lld\n", dp[N]);
return 1;//防抄
}