CodeForces 1453 B. Suffix Operations

CodeForces 1453 B. Suffix Operations

题目大意：

给定一个数组，每次操作可以选择一个后缀全部加一或者全部减一，你有一次机会改变一个数字为任意的，问最少需要多少次操作能让数组元素全部一样。

思路：

因为每次只能操作后缀，就很自然想到从后开始操作，想让 $a_n$ 等于 $a_{n-1}$, 然后等于 $a_{n-2}$， 一直下去，算一个后缀和，这就是不用机会的最小值。之后枚举每一个位置，对于中间 $ 1 < i < n$ 的位置，减少的次数其实就是 $|a_{i-1}-a_i|+|a_i-a_{i+1}|-|a_{i-1}-a_{i+1}|$ , 对于头尾特殊判断一下即可。

代码：

#include <bits/stdc++.h>
using namespace std;
#define me(a, b) memset(a, b, sizeof(a))
#define IOS() ios::sync_with_stdio(false), cin.tie(0)
#define endl '\n'

typedef long long ll;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;
const int INF = 0x3f3f3f3f;
const int maxn = 2e5 + 5;
const ll mod = 1e9 + 7;

ll a[maxn];
ll suff[maxn];

void solve()
{
    int n;
    cin >> n;
    for(int i = 0; i < n; ++i) {
        cin >> a[i];
    }
    if(n <= 2) {
        cout << 0 << endl;
        return ;
    }
    a[n] = a[n-1];
    suff[n-1] = suff[n] = 0;

    for(int i = n-2; i >= 0; --i) {
        suff[i] = suff[i+1] + abs(a[i]-a[i+1]);
    }
    ll sum = suff[0];
    ll ans = min(suff[1], sum - suff[n-2]);
    for(int i = 1; i < n-1; ++i) {
        ll x = abs(a[i-1] - a[i+1]);
        ll y = abs(a[i-1] - a[i]) + abs(a[i] - a[i+1]);;
        ans = min(ans, sum-y+x);
    }
    
    cout << ans << endl;
}

int main()
{
    IOS();
    int T;
    cin >> T;
    while(T--) {
        solve();
    }
    return 0;
}

总结：

想的慢了，边界的细节写太久了