题目:BZOJ1367.
题目大意:给定一个长度为
n
n
n的序列
t
t
t,要求构造一个严格递增的序列
z
z
z,使得
s
=
∑
i
=
1
n
∣
z
i
−
t
i
∣
s=\sum_{i=1}^{n}|z_i-t_i|
s=∑i=1n∣zi−ti∣最小,并输出这个
s
s
s.
1
≤
n
≤
1
0
6
1\leq n\leq 10^6
1≤n≤106.
跟着道题差不多的还有三道题:POJ3666,CF713C,hihocoder1529.
首先考虑把 t i t_i ti减去 i i i得到一个新的序列 t t t,那么原问题的严格递增就可以变为非严格递增.
考虑一个性质,必定存在一个最优的序列 z z z使得 z z z中所有出现过的数是 t t t中所有出现过的数的子集.
证明是不存在的,这很显然嘛…
于是我们可以设
f
[
i
]
f[i]
f[i]表示前
i
i
i个数,其中满足
z
[
i
]
=
t
[
i
]
z[i]=t[i]
z[i]=t[i]时最小的
s
s
s.很容易得到方程:
f
[
i
]
=
min
j
=
0
i
−
1
{
f
[
j
]
+
∑
k
=
j
+
1
i
∣
t
i
−
t
k
∣
}
f[i]=\min_{j=0}^{i-1}\{f[j]+\sum_{k=j+1}^{i}|t_i-t_k|\}
f[i]=j=0mini−1{f[j]+k=j+1∑i∣ti−tk∣}
这个做法是 O ( n 3 ) O(n^3) O(n3)的,过不了此题.
所以考虑换一个状态,设
f
[
i
]
[
j
]
f[i][j]
f[i][j]表示前
i
i
i个数,其中
z
i
=
j
z_i=j
zi=j时最小的
s
s
s.容易得到方程:
f
[
i
]
[
j
]
=
min
k
=
0
j
{
f
[
i
−
1
]
[
k
]
+
∣
t
i
−
j
∣
}
f[i][j]=\min_{k=0}^{j}\{f[i-1][k]+|t_i-j|\}
f[i][j]=k=0minj{f[i−1][k]+∣ti−j∣}
直接做的复杂度仍然是 O ( n 2 ) O(n^2) O(n2)的,依旧过不了此题.
考虑一下把
f
[
i
]
f[i]
f[i]看成关于
j
j
j的函数,容易通过数学归纳法证明图像是具有单峰性的,其中峰值为最小值:
如果是通过这个状态得到答案,答案为
min
i
=
0
t
n
{
f
[
n
]
[
i
]
}
\min_{i=0}^{t_n}\{f[n][i]\}
mini=0tn{f[n][i]},虽然这个函数具有凹凸性,但这并不好维护.
我们再设
F
[
i
]
[
j
]
F[i][j]
F[i][j]为
f
[
i
]
[
j
]
f[i][j]
f[i][j]的前缀最小值,也就是说
F
[
i
]
[
j
]
F[i][j]
F[i][j]表示
z
i
≤
j
z_i\leq j
zi≤j时最小的
s
s
s.容易得到另一个方程:
F
[
i
]
[
j
]
=
min
k
=
0
j
{
F
[
i
−
1
]
[
k
]
+
∣
t
i
−
k
∣
}
=
min
k
=
0
j
{
f
[
i
]
[
j
]
}
F[i][j]=\min_{k=0}^{j}\{F[i-1][k]+|t_i-k|\}=\min_{k=0}^{j}\{f[i][j]\}
F[i][j]=k=0minj{F[i−1][k]+∣ti−k∣}=k=0minj{f[i][j]}
现在答案就变成了简单的一个值
F
[
n
]
[
t
n
]
F[n][t_n]
F[n][tn],并且容易发现
F
[
i
]
F[i]
F[i]的图像就会呈现一个很优秀的单调性:
我们考虑从从
f
[
i
]
f[i]
f[i]转移到
f
[
i
+
1
]
f[i+1]
f[i+1]有什么变化,发现无非多了一个绝对值函数
∣
t
i
−
j
∣
|t_i-j|
∣ti−j∣:
那么这一个增加的量对
f
f
f的函数图像会产生什么效果呢?显然是对所有横坐标在
t
i
t_i
ti前的线段斜率
−
1
-1
−1,在
t
i
t_i
ti后的线段斜率
+
1
+1
+1.
要维护这个东西可不好搞啊…考虑这个玩意会对 F F F的图像造成什么影响.
设
f
[
i
]
f[i]
f[i]图像最低点的横坐标为
x
i
x_i
xi,则对于
F
[
i
]
⇒
F
[
i
+
1
]
F[i] \Rightarrow F[i+1]
F[i]⇒F[i+1]我们需要分两类讨论:
1.若
x
i
≤
t
i
+
1
x_i\leq t_{i+1}
xi≤ti+1,容易发现在
t
i
+
1
t_{i+1}
ti+1之前的所有线段斜率
−
1
-1
−1,而
t
i
+
1
t_{i+1}
ti+1之后的线段斜率仍然为
0
0
0,纵坐标为
F
[
i
]
[
x
i
]
F[i][x_i]
F[i][xi].用图像来表示就是:
2.若
x
i
>
t
i
+
1
x_i>t_{i+1}
xi>ti+1,容易发现现在还是给
t
i
+
1
t_{i+1}
ti+1前的线段斜率
−
1
-1
−1,但是只给
t
i
+
1
t_{i+1}
ti+1后斜率小于
0
0
0的线段斜率
+
1
+1
+1,等于
0
0
0的依旧为
0
0
0,用图像来表示就是:
虽然我知道这个例子不是很好,但是我懒得换了QAQ…
然后这个变换会使得图像中的最低点的纵坐标也就是答案增加了多少呢?不难发现一定是最左边的最低点的横坐标减去 t i + 1 t_{i+1} ti+1,也就是一定 f [ i + 1 ] [ x i + 1 ] f[i+1][x_{i+1}] f[i+1][xi+1]一定由 f [ i ] [ x i ] f[i][x_i] f[i][xi]转移过去.
为什么呢?可以观察一下 F [ i ] F[i] F[i]的图像,发现 x i x_i xi之前的线段斜率一定小于等于 − 1 -1 −1,而 + 1 +1 +1后这个点必然还是最低点(只是不是最左边的了),所以这样转移一定最优.
知道了这些后,我们该干什么呢?
首先 F [ i ] F[i] F[i]图像上所有线段的斜率必然是在区间 [ − 2 i , 0 ] [-2i,0] [−2i,0]上的整数,这可以通过上面的斜率变换得到.
现在我们只需要维护每一条斜率不同的线段即可,但是显然这样很麻烦,考虑只维护每一个不同斜率的左端点的横坐标.
只需要对横坐标维护一个优先队列即可,可以通过这个优先队列维护我们上面需要的所有操作,具体实现可以看代码.
时间复杂度 O ( n log n ) O(n\log n) O(nlogn).
代码如下:
#include<bits/stdc++.h>
using namespace std;
#define Abigail inline void
typedef long long LL;
priority_queue<int>q;
int n;
LL ans;
Abigail into(){
scanf("%d",&n);
int a;
for (int i=1;i<=n;++i){
scanf("%d",&a);a-=i;
q.push(a);
if (a<q.top()) ans+=(LL)q.top()-a,q.pop(),q.push(a);
}
}
Abigail outo(){
printf("%lld\n",ans);
}
int main(){
into();
outo();
return 0;
}