题意分析
将数组 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,随后可以选择左边的元素或右边的元素进行同样操作。
根据贪心的思想,如果第一个元素需要执行操作,那么我们可以看下一个元素是否需要同步进行操作,如果操作后对结果更优则执行,反之跳过。
import sys
sys.setrecursionlimit(1000000)
input = lambda:sys.stdin.readline().strip()
n = int(input())
a = list(map(int, input().split()))
d = []
for i in range(n // 2):
d.append(a[i] - a[n - i - 1])
n = len(d)
res = 0
for i in range(n):
res += abs(d[i])
if i != n - 1:
if d[i] > 0 and d[i + 1] > 0:
d[i + 1] -= min(d[i], d[i + 1])
elif d[i] < 0 and d[i + 1] < 0:
d[i + 1] -= max(d[i], d[i + 1])
print(res)