西南交大峨眉校区INT杯2020-I.增减序列【差分+贪心】

思路

差分的经典好题

前置知识：差分
将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;
}