思路
差分的经典好题
前置知识:差分
将a数组求一个差分(b[i] = a[i] - a[i - 1]),我们会得到一个b[i]数组,我们要将某个区间+1或者-1操作,就是对b数组中的b[i]–,b[j]++操作(1<=i,j<=n)。这儿分类讨论一下
如果i-1>j,则是在j到i-1这个区间+1;
如果j-1>i,则是在i到j-1这个区间-1;
要让a数组变成全部相同的数,等价于将b数组全部变成0. 假设b[i]为正数,b[j]为负数,我们尽量让b[i]–,b[j]++,这样就是最优了吗,其实不然,要是正负数对数不相等怎么办,这就要将b数组剩下没变成0的数与b[1]或者b[n + 1]进行+1或者-1操作,改变b[1]将会影响到a[1], 而改变b[n+1]则不会影响到任何部分。
最小步数:min(pos,neg) + abs(pos-neg)。
变成相同的数:最终b[1] (+1或者-1操作)改变的情况有:0, 1, 2, …abs(pos-neg); 一共abs(pos-neg)个。
这儿最后的方案数可以这样理解。
如果b[1] 改变0次的情况,则是一直在改变b[n+1],
如果b[1]改变1次的情况,则是改变了(abs(pos-neg)-1)次。
依次类推。
AC代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 1e5 + 5;
int a[MAXN], dif[MAXN];
void solve() {
int n;
scanf("%d", &n);
for (int i = 1; i <= n; ++i) {
scanf("%d", &a[i]);
}
int pos = 0, neg = 0;
for (int i = 2; i <= n; ++i) {
dif[i] = a[i] - a[i - 1];
if (dif[i] > 0) {
pos += dif[i];
} else {
neg -= dif[i];
}
}
printf("%d\n%d\n", min(pos, neg) + abs(pos - neg), abs(pos - neg) + 1);
}
int main() {
solve();
return 0;
}