首先,给你一个初始数组 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
思路:
直接模拟即可。
注意每次修改数组时要在新开的数组上修改,不然已经修改的值会对随后的修改产生影响。
时间复杂度: O(KN), K是循环的次数,N是数组长度。每次循环都需要遍历整个数组。
空间复杂度: O(N),因为开辟了额外的数组用于储存临时结果。
class Solution(object):
def transformArray(self, arr):
"""
:type arr: List[int]
:rtype: List[int]
"""
flag = 1
while flag:
flag = 0 # 如果没有修改发生,则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