CodeForces 1453 B. Suffix Operations
题目大意:
给定一个数组,每次操作可以选择一个后缀全部加一或者全部减一,你有一次机会改变一个数字为任意的,问最少需要多少次操作能让数组元素全部一样。
思路:
因为每次只能操作后缀,就很自然想到从后开始操作,想让 a n a_n an 等于 a n − 1 a_{n-1} an−1, 然后等于 a n − 2 a_{n-2} an−2, 一直下去,算一个后缀和,这就是不用机会的最小值。之后枚举每一个位置,对于中间 $ 1 < i < n$ 的位置,减少的次数其实就是 ∣ a i − 1 − a i ∣ + ∣ a i − a i + 1 ∣ − ∣ a i − 1 − a i + 1 ∣ |a_{i-1} - a_{i}| + |a_{i} - a_{i+1}| - |a_{i-1} - a_{i+1}| ∣ai−1−ai∣+∣ai−ai+1∣−∣ai−1−ai+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;
}
总结:
想的慢了,边界的细节写太久了