leetcode_1243 数组变换

要求

首先，给你一个初始数组 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]。
无法再对该数组进行更多操作。

核心代码

class Solution:
    def transformArray(self, arr: List[int]) -> List[int]:
        flag = 1
        while flag:
            flag = 0
            res = [num for num in arr]
            for i in range(1,len(arr) - 1):
                if arr[i - 1] < arr[i] and arr[i] > arr[i + 1]:
                    res[i] -= 1
                    flag = 1
                elif arr[i - 1] > arr[i] and arr[i] < arr[i + 1]:
                    res[i] += 1
                    flag = 1
            arr = res[:]
        return res 

解题思路：我们使用标志位对数组的改变进行标志，当没有改变的时候，使用标志位可以将程序退出来，加减1来自题目的规则，把握好题目规则即可，比较简单。