题意
给定一个长度为 n 的数列
a
[
]
a[]
a[],一共有三种操作:
- 选择一个位置 i,令前 i 个位置上的数 -1;
- 选择一个位置 i,令最后 i 个位置上的数 -1;
- 所有位置上的数 +1。
问,最少多少次操作能把所有数变为 0?
1 ≤ n ≤ 200 000 1 \leq n \leq 200\,000 1≤n≤200000
思路
对于区间位置,要能习惯性的转化为差分。
预处理出差分数组
d
[
]
d[]
d[]。
对于操作 1,转化为 d[0] -= 1,d[i+1] += 1
;
对于操作 2,转化为 d[i] -= 1,d[n+1] += 1,而对于n+1位置不用考虑,所以只有 d[i] -= 1
;
对于操作 3,转化为 d[1] += 1,d[n+1] -=1,同样 n+1 位置不用考虑,只有 d[1] += 1
。
发现如果让一个 >1 的位置上的数增加,只能用 操作 1;如果让一个 >1 的位置上的数减小,只能用操作 2。把 >1 的所有位置都变成 0 之后,再用操作 2 和 操作 3 处理 1 位置即可。
#include<bits/stdc++.h>
using namespace std;
#define Ios ios::sync_with_stdio(false),cin.tie(0)
const int N = 200010, mod = 1e9+7;
int T, n, m;
int a[N], d[N];
signed main(){
Ios;
cin >> T;
while(T--)
{
cin >> n;
int sum = 0;
for(int i=1;i<=n;i++){
cin >> a[i];
d[i] = a[i] - a[i-1];
if(i > 1 && d[i] < 0) sum += -d[i];
}
d[1] -= sum;
if(d[1] < 0){
sum += -d[1];
d[1] = 0;
}
for(int i=1;i<=n;i++) if(d[i] > 0) sum += d[i];
cout << sum << endl;
}
return 0;
}
场上完全没有想到用差分。。