LeetCode 1243. 数组变换

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

---

长按或扫码关注我的公众号，一起加油、一起学习进步！