当前位置： 移动技术网 > IT编程>脚本编程>Python > python3数据结构算法（数组变换）

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

2020年07月04日  | 移动技术网IT编程  | 我要评论


文章目录


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

您可能感兴趣的文章:

如对本文有疑问， 点击进行留言回复！！

相关文章:

验证码：
最近更新的文章
大家感兴趣的文章
移动技术网