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:# 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
长按或扫码关注我的公众号,一起加油、一起学习进步!