README.md

1243. 数组变换

English Version

题目描述

首先，给你一个初始数组 arr。然后，每天你都要根据前一天的数组生成一个新的数组。

第 i 天所生成的数组，是由你对第 i-1 天的数组进行如下操作所得的：

1. 假如一个元素小于它的左右邻居，那么该元素自增 1。
2. 假如一个元素大于它的左右邻居，那么该元素自减 1。
3. 首、尾元素 永不 改变。

过些时日，你会发现数组将会不再发生变化，请返回最终所得到的数组。

 

示例 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. 1 <= arr.length <= 100
2. 1 <= arr[i] <= 100

解法

Python3

class Solution:
    def transformArray(self, arr: List[int]) -> List[int]:
        copy = [e for e in arr]
        has_change, n = True, len(arr)
        while has_change:
            has_change = False
            for i in range(1, n - 1):
                if arr[i] < copy[i - 1] and arr[i] < copy[i + 1]:
                    arr[i] += 1
                    has_change = True
                elif arr[i] > copy[i - 1] and arr[i] > copy[i + 1]:
                    arr[i] -= 1
                    has_change = True
            copy = [e for e in arr]
        return arr

Java

class Solution {
    public List<Integer> transformArray(int[] arr) {
        int n = arr.length;
        int[] copy = Arrays.copyOf(arr, n);
        boolean hasChange = true;
        while (hasChange) {
            hasChange = false;
            for (int i = 1; i < n - 1; ++i) {
                if (arr[i] < copy[i - 1] && arr[i] < copy[i + 1]) {
                    ++arr[i];
                    hasChange = true;
                } else if (arr[i] > copy[i - 1] && arr[i] > copy[i + 1]) {
                    --arr[i];
                    hasChange = true;
                }
            }
            System.arraycopy(arr, 0, copy, 0, n);
        }
        List<Integer> res = new ArrayList<>();
        for (int e : arr) {
            res.add(e);
        }
        return res;
    }
}

...