LeetCode 1243. 数组变换

文章目录

- 1. 题目
- 2. 解题

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/array-transformation
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

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:# py3def 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] += 1elif arr[i-1] < arr[i] and arr[i] > arr[i+1]:temp[i] -= 1if temp == arr:breakarr = temp.copy()return arr

52 ms 13.6 MB

class Solution: # py3def 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] += 1elif arr[i-1] < arr[i] and arr[i] > arr[i+1]:temp[i] -= 1if temp == arr:breakarr = temp[:]return arr