题目
题目描述
P 教授要去看奥运,但是他舍不下他的玩具,于是他决定把所有的玩具运到北京。他使用自己的压缩器进行压缩,其可以将任意物品变成一堆,再放到一种特殊的一维容器中。
P 教授有编号为 1 \cdots n1⋯n 的 nn 件玩具,第 ii 件玩具经过压缩后的一维长度为 C_iC
i
。
为了方便整理,P教授要求:
在一个一维容器中的玩具编号是连续的。
同时如果一个一维容器中有多个玩具,那么两件玩具之间要加入一个单位长度的填充物。形式地说,如果将第 ii 件玩具到第 jj 个玩具放到一个容器中,那么容器的长度将为 x=j-i+\sum\limits_{k=i}^{j}C_kx=j−i+
k=i
∑
j
C
k
。
制作容器的费用与容器的长度有关,根据教授研究,如果容器长度为 xx,其制作费用为 (x-L)^2(x−L)
2
。其中 LL 是一个常量。P 教授不关心容器的数目,他可以制作出任意长度的容器,甚至超过 LL。但他希望所有容器的总费用最小。
输入格式
第一行有两个整数,用一个空格隔开,分别代表 nn 和 LL。
第 22 到 第 (n + 1)(n+1) 行,每行一个整数,第 (i + 1)(i+1) 行的整数代表第 ii 件玩具的长度 C_iC
i
。
输出格式
输出一行一个整数,代表所有容器的总费用最小是多少。
输入输出样例
输入 #1 复制
5 4
3
4
2
1
4
输出 #1 复制
1
说明/提示
对于全部的测试点,1 \leq n \leq 5 \times 10^41≤n≤5×10
4
,1 \leq L \leq 10^71≤L≤10
7
,1 \leq C_i \leq 10^71≤C
i
≤10
7
。
思路
一看就知道是要DP,状态转移方程一写
dp[i]=min(dp[j]+(sum[i]+i−sum[j]−j−L−1)^2)(j<i)
那么如果我们要去掉min
就变成了这样
dp[i]=dp[j]+(sum[i]+i−sum[j]−j−L−1)^2(j<i)
设sum[i]+i=a[i],sum[i]+i+L+1=b[i]
那么dp[i]=dp[j]+a[i]2+b[j]2-2a[i]b[j]
如果式子长这样:
dp[i]=max(dp[j]+a[j])(只是类似)
那么我们可以使用单调队列优化
但是我们的这个式子中有一项为i,j相乘!!!
怎么办?凉了吗?
这时斜率优化横空而出
移一下项,有:
dp[j]+b[j]2=2a[i]b[j]+dp[i]-a[i]2
等一下,这里i为定值,同时为需要转移状态,所以除了dp[i]之外所有与i相关的项为定值
我们需要dp[i]最小,那么
dp[j]+b[j]2(y)=2a[i](k)b[j](x)+dp[i]-a[i]2(b)
形如y=kx+b(废话)
考虑平面内一系列的点(即每个决策点)
当前仅当形如上图的下凸包的点才可能成为最优决策
那么我们用单调队列维护一下
就是要保证一个点,其左边斜率小于右边斜率即可
队首出队条件当队首斜率小于当前直线斜率(2*a[i])
主要原因是因为2*a[i]单调,但有些题目中不一定
那我们就必须要二分在其中查找最优决策
代码
#include<bits/stdc++.h>
using namespace std;
int read() {
int res = 0; bool bo = 0; char c;
while (((c = getchar()) < '0' || c > '9') && c != '-');
if (c == '-') bo = 1; else res = c - 48;
while ((c = getchar()) >= '0' && c <= '9')
res = (res << 3) + (res << 1) + (c - 48);
return bo ? ~res + 1 : res;
}
typedef long long ll;
const int N = 5e4 + 5;
int n, H, T, L, que[N];
ll f[N], a[N], b[N], X[N], Y[N];
bool check(int p1, int p2, int p3) {
return (X[p2] - X[p3]) * (Y[p1] - Y[p3]) - (X[p1] - X[p3])
* (Y[p2] - Y[p3]) >= 0;
}
ll calc(int x, int y)
{
return f[x] + b[x] * b[x] - 2ll * a[y] * b[x];
}
int main()
{
int i; n = read(); L = read(); H = T = 1;
for (i = 1; i <= n; i++) {
b[i] = b[i - 1] + read() + 1;
a[i] = b[i] - L - 1;
}
for (i = 1; i <= n; i++) {
while (H < T && calc(que[H], i) >= calc(que[H + 1], i)) H++;
f[i] = calc(que[H], i) + a[i] * a[i];
X[i] = b[i]; Y[i] = f[i] + b[i] * b[i];
while (H < T && check(que[T - 1], que[T], i)) T--;
que[++T] = i;
}
cout << f[n] << endl;
return 0;
}