P4552 [Poetize6] 差分

题意

传送门 P4552 [Poetize6] IncDec Sequence

题解

求出 $a$ 的差分序列 $d$，其中 $d_0=a_0,d_i=a_i-a_{i-1}(1\le i<n)$。令 $a_n=0$，那么对区间的操作转化为对序列中两个数的加或减 $1$ 操作，目标是使 $d_1,d_2\dots ,d_{n-1}$ 为 $0$，此时序列由 $n$ 个 $d_0$ 组成。对区间 $[i,j]$ 进行操作有 $3$ 种可能的可以逼近目标的情况：取 $b_i,b_j(1\le i,j<n)$，此时单次的操作可以最快地逼近目标，所以应该尽量采取这样的操作；取 $b_0,b_j(1\le j<n)$；取 $b_i,b_n(1\le i<n)$。

设 $d_1,d_2\dots ,d_{n-1}$ 中正数的和为 $x$，负数的和为 $y$。第一种情况最多可执行 $min(x,y)$ 次操作，第二、三种情况可执行 $|x-y|$ 次操作；总共 $max(x,y)$ 次，即最小操作次数。

对差分序列 $d$ 求前缀和，可得到最终序列，即由 $n$ 个 $d_0$ 组成的序列。那么不同的序列数等于 $d_0$ 可能的取值数量，根据上述分析，有 $d_0\in [0,|x-y|]$，故不同序列数为 $|x-y|+1$。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define maxn 100005
int N;
ll A[maxn], D[maxn];

int main()
{
    scanf("%d", &N);
    for (int i = 0; i < N; ++i)
        scanf("%lld", A + i);
    D[0] = A[0];
    ll x = 0, y = 0;
    for (int i = 1; i < N; ++i)
    {
        D[i] = A[i] - A[i - 1];
        if (D[i] < 0)
            y -= D[i];
        else
            x += D[i];
    }
    ll res = max(x, y), num = abs(x - y) + 1;
    printf("%lld\n%lld\n", res, num);
}