python3数据结构算法（数组变换）

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 { //C++ public: vector<int> transformArray(vector<int>& arr) { vector<int> temp(arr); int i, n = arr.size(); while(1) { for(i = 1; i < n-1; ++i) { if(arr[i-1] > arr[i] && arr[i] < arr[i+1]) temp[i]++; else if(arr[i-1] < arr[i] && arr[i] > arr[i+1]) temp[i]--; } if(temp == arr) break; arr = temp; } return arr; } }; 

4 ms 7.3 MB

python3 解答，注意使用 copy，创建副本，或者使用 list[:]生成副本

class Solution:# py3 def transformArray(self, arr: List[int]) -> List[int]: temp = arr.copy() while True: for i in range(1, len(arr)-1): if arr[i-1] > arr[i] and arr[i] < arr[i+1]: temp[i] += 1 elif arr[i-1] < arr[i] and arr[i] > arr[i+1]: temp[i] -= 1 if temp == arr: break arr = temp.copy() return arr 

52 ms 13.6 MB

class Solution: # py3 def transformArray(self, arr: List[int]) -> List[int]: temp = arr[:] while True: for i in range(1, len(arr)-1): if arr[i-1] > arr[i] and arr[i] < arr[i+1]: temp[i] += 1 elif arr[i-1] < arr[i] and arr[i] > arr[i+1]: temp[i] -= 1 if temp == arr: break arr = temp[:] return arr 

本文地址：https://blog.csdn.net/qq_21201267/article/details/107097416