【ACWing】100. 增减序列

题目地址：

https://www.acwing.com/problem/content/102/

给定一个长度为$n$的数列$a_1,a_2,\dots ,a_n$，每次可以选择一个区间$[l,r]$，使下标在这个区间内的数都加一或者都减一。求至少需要多少次操作才能使数列中的所有数都一样，并求出在保证最少次数的前提下，最终得到的数列可能有多少种。

输入格式：
第一行输入正整数$n$。
接下来$n$行，每行输入一个整数，第$i+1$行的整数代表$a_i$。

输出格式：
第一行输出最少操作次数。
第二行输出最终能得到多少种结果。

数据范围：
$0<n\le 10^5$
$0\le a_i<2147483648$

首先求一下$a$的差分数组$b$，即$b_i=a_i-a_{i-1}$。那么将某个区间$[l,r]$的数都加$1$等价于将$b[l]$加$1$并且将$b[r+1]$减$1$；而如果将某个区间$[l,r]$的数都减$1$等价于将$b[l]$减$1$并且将$b[r+1]$加$1$。我们首先考虑最少操作次数应该是多少。如果最后真的让所有的数都变成了一样，那么$b$会满足$b_2=b_3=...=b_n=0$。我们不妨设${\sum }_{b_i>0}{b}_i=p,-{\sum }_{b_i<0}{b}_i=q$，如果$p>q$，那么我们可以先每次都从$b_{2\sim n+1}$中选取一正一负，正的减$1$，负的加$1$，先把$b$中的负数全消灭掉，接下来如果还有非零的数，不妨设这些数都是负的，这样可以通过调整$b_1$和$b_{n+1}$的办法将这些负数也都变为$0$，最终就达到了将$b$调整为满足条件的序列的结果。这种操作方式的次数为 min ⁡ { p , q } + ∣ p − q ∣ \min\{p,q\}+|p-q| min{p,q}+∣p−q∣，并且一定是次数最少的，因为要将$b_{2\sim n}$都变为$0$的话，正负抵消一定是最快的，抵消不掉的则需要借助$b_1$。最后得到的数列的个数即取决于$b_1$变成了什么，因为正负抵消掉之后的调整过程，可以调整$b_1$也可以调整$b_{n+1}$，所以一共有$|p-q|+1$种可能。代码如下：

#include <iostream>
using namespace std;
using ll = long long;

const int N = 1e5 + 10;
int a[N];

int main() {
  int n;
  scanf("%d", &n);
  for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
  for (int i = n; i > 1; i--) a[i] -= a[i - 1];

  ll pos = 0, neg = 0;
  for (int i = 2; i <= n; i++)
    if (a[i] > 0) pos += a[i];
    else neg -= a[i];
  
  printf("%lld\n", min(pos, neg) + abs(pos - neg));
  printf("%lld\n", abs(pos - neg) + 1);
}

时空复杂度$O(n)$。