比赛时只想到了 O ( n 3 ) \mathcal O(n^3) O(n3) 的暴力做法,官方题解是 O ( n 2 ) \mathcal O(n^2) O(n2) ,并且是可以优化为 O ( n ) \mathcal O(n) O(n) 的(贪心+差分)。
贪心是比较明显的,我们需要从左到右依次考虑每张床的贡献,因为第 i i i 张床只能由 [ 1 , i − 1 ] [1,i-1] [1,i−1] 的床跳过来,Pekora从第一张床开始跳一定是最优的。我们从左到右分析,用 b i b_i bi 记录在 i i i 床在之前被跳过的次数,如果 b i > S i − 1 b_i>S_i-1 bi>Si−1 ,说明 S i S_i Si 在之前就已经降为 1 1 1 了,多跳的部分会转移到 b i + 1 b_{i+1} bi+1 上;否则还需要在 i i i 床上跳 S i − b i − 1 S_i-b_i-1 Si−bi−1 次。然后对于每张床我们只需要考虑它能直接跳到的地方:因为它要从 S i S_i Si 减到 1 1 1 ,所以它会把 [ i + 2 , m i n ( S i + i , n ) ] [i+2,min(S_i+i,n)] [i+2,min(Si+i,n)] 区间上的所有床都跳过一次,即这个区间上的 b b b 都增一,这里如果用差分的话就是 O ( 1 ) \mathcal O(1) O(1) 的区间修改复杂度了,当然用暴力扫一遍也是可以过这道题的。这里给出使用差分的代码, b b b 在代码中为差分数组。
C o d e : \mathrm{Code}: Code:
#include<bits/stdc++.h>
using namespace std;
const int N = 1e4 + 6;
typedef long long ll;
int n,s[N],b[N];//注意b[i]为差分数组
int main(){
int t;scanf("%d",&t);
while(t--){
scanf("%d",&n);
ll ans = 0;
for(int i = 1;i <= n;i++) scanf("%d", s + i), b[i] = 0;b[n + 1]=0;
int las = 0;
for(int i = 1;i <= n;i++){
int now = b[i] + las; //now表示是i床在[1,i-1]中被跳了几次,即上面分析中的b[i]
las = now;
if(now > s[i] - 1){
b[i+1] += now - s[i] + 1;
b[i+2] -= now - s[i] + 1;
now = s[i] - 1;
}
ans += s[i] - now - 1;
if(min(s[i] + i, n) >= i + 2){
b[i+2] ++; //相当于区间[i+2,min(s[i]+i,n)]增一
b[min(s[i] + i, n) + 1]--;
}
}
cout<<ans<<endl;
}
return 0;
}