跳转至

LeetCode: 1243. 数组变换

1、题目描述

首先,给你一个初始数组 arr。然后,每天你都要根据前一天的数组生成一个新的数组。

i 天所生成的数组,是由你对第 i-1 天的数组进行如下操作所得的:

假如一个元素小于它的左右邻居,那么该元素自增 1。 假如一个元素大于它的左右邻居,那么该元素自减 1。 首、尾元素 永不 改变。 过些时日,你会发现数组将会不再发生变化,请返回最终所得到的数组。

示例 1:

输入:[6,2,3,4]
输出:[6,3,3,4]
解释:
第一天,数组从 [6,2,3,4] 变为 [6,3,3,4]。
无法再对该数组进行更多操作。

示例 2:

输入:[1,6,3,4,3,5]
输出:[1,4,4,4,4,5]
解释:
第一天,数组从 [1,6,3,4,3,5] 变为 [1,5,4,3,4,5]。
第二天,数组从 [1,5,4,3,4,5] 变为 [1,4,4,4,4,5]。
无法再对该数组进行更多操作。

提示:

  • 1 <= arr.length <= 100
  • 1 <= arr[i] <= 100

2、解题思路

  • 直接模拟即可,注意不要在原数组修改
class Solution:
    def transformArray(self, arr: List[int]) -> List[int]:
        length = len(arr)
        if length <= 2:
            return arr

        ans = arr[:]
        flag = True

        while flag:

            for i in range(1, length - 1):
                if arr[i - 1] < arr[i] and arr[i + 1] < arr[i]:
                    ans[i] = arr[i] - 1
                    flag = False
                elif arr[i - 1] > arr[i] and arr[i + 1] > arr[i]:
                    flag = False
                    ans[i] = arr[i] + 1
                else:
                    ans[i] = arr[i]

            flag = not flag
            if flag:
                ans, arr = arr, ans

        return arr