算法竞赛进阶指南打卡系列题解 0x00 13.增减序列

算法竞赛进阶指南打卡系列题解

---

题目

原题链接

AcWing 100. IncDec序列

题面

给定一个长度为 $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$

输入样例：

4
1
1
2
2

输出样例：

1
2

题解

思路

对于区间进行加减操作，肯定是差分。

因为最终序列需要满足全部相等，因此我们只需要使得差分数组第一项为定值，$[2,n]$ 项为 $0$ 即可。

每次操作使得区间加一或减一，对应到差分数组中即为使得一个数减一一个数加一，因此操作数即为差分数组 $[2,n]$ 项中正数和与负数和绝对值较大的那个，因为每次操作我们均可将其分别作用在一个负数和一个整数上，只有这样操作数才最小。

对于结果个数，我们假设此时正数和为 $p$，负数和的绝对值为 $q$，$p>q$，经过 $q$ 次操作后，我们还需要减去 $p-q$ 个 $1$，而对于加上的部分，我们可以加在第 $n+1$ 项上，也可以加在第 $1$ 项上，只有加在第 $1$ 项上才对结果有影响，不同的分配数一共有 $p-q+1$ 种，因此答案为 $abs(p-q)+1$。

代码

#include <bits/stdc++.h>

// #define int long long

using namespace std;

constexpr int P = 998244353;
using i64 = long long;

const int N = 100010;

int n;
int a[N], b[N];

signed main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    cin >> n;
    for (int i = 1; i <= n; i ++ ) cin >> a[i];

    for (int i = 1; i <= n; i ++ ) {
        b[i] += a[i];
        b[i + 1] -= a[i];
    }

    i64 p = 0, q = 0;
    for (int i = 2; i <= n; i ++ ) {
        if (b[i] > 0) p += b[i];
        else q -= b[i];
    }

    cout << max(p, q) << "\n" << abs(p - q) + 1 << "\n";

    return 0;
}