题意分析
将数组 a a a 通过最少的操作次数转换为一个回文数组:
- 操作 1 1 1:将任意相邻的两个数一起 + 1 +1 +1 或者 − 1 -1 −1。
- 操作 2 2 2:将任意一个数 + 1 +1 +1 或者 − 1 -1 −1。
解题思路
回文数组:满足条件 a i = a n − i − 1 ( i ∈ [ 0 , n − 1 ] ) a_i = a_{n-i-1} (i\in[0, n - 1]) ai=an−i−1(i∈[0,n−1])。
即数组中的第一个元素,要等于最后一个元素,第二个元素,要等于倒二个元素, … \dots …
那么,假设操作第一个元素和第二个元素 + 1 +1 +1,相当于操作倒数两个元素 − 1 -1 −1,因为这样对操作次数是不影响的。
所以我们的目标转换为,操作前一半元素,使得数组变为回文数组。
我们可以计算出前一半元素只执行操作 2 2 2 使数组变为回文数组的操作次数。
因为操作 1 1 1 可以看作先执行一次操作 2 2 2,随后可以选择左边的元素或右边的元素进行同样操作。
根据贪心的思想,如果第一个元素需要执行操作,那么我们可以看下一个元素是否需要同步进行操作,如果操作后对结果更优则执行,反之跳过。
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <vector>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int n;
int a[N];
vector<int> p;
int main()
{
cin >> n;
for (int i = 0; i < n; ++ i)
cin >> a[i];
for (int i = 0; i < n / 2; ++ i )
p.push_back(a[i] - a[n - i - 1]);
n = p.size();
LL res = 0;
for (int i = 0; i < n; ++ i )
{
res += abs(p[i]);
if (i != n - 1)
{
if (p[i] > 0 && p[i + 1] > 0)
p[i + 1] -= min(p[i], p[i + 1]);
else if (p[i] < 0 && p[i + 1] < 0)
p[i + 1] -= max(p[i], p[i + 1]);
}
}
cout << res << endl;
return 0;
}