Formatted question description: https://leetcode.ca/all/1243.html

1243. Array Transformation

Level

Easy

Description

Given an initial array arr, every day you produce a new array using the array of the previous day.

On the i-th day, you do the following operations on the array of day i-1 to produce the array of day i:

  1. If an element is smaller than both its left neighbor and its right neighbor, then this element is incremented.
  2. If an element is bigger than both its left neighbor and its right neighbor, then this element is decremented.
  3. The first and last elements never change.

After some days, the array does not change. Return that final array.

Example 1:

Input: arr = [6,2,3,4]

Output: [6,3,3,4]

Explanation:

On the first day, the array is changed from [6,2,3,4] to [6,3,3,4].

No more operations can be done to this array.

Example 2:

Input: arr = [1,6,3,4,3,5]

Output: [1,4,4,4,4,5]

Explanation:

On the first day, the array is changed from [1,6,3,4,3,5] to [1,5,4,3,4,5].

On the second day, the array is changed from [1,5,4,3,4,5] to [1,4,4,4,4,5].

No more operations can be done to this array.

Constraints:

  • 1 <= arr.length <= 100
  • 1 <= arr[i] <= 100

Solution

Do the operations on the array directly. On each day, calculate whether the numbers at each index in the array should be incremented or decremented or remain unchanged, and do the modification to the array accordingly. If on one day the array is unchanged, then return the final array, which needs to be converted to a list.

class Solution {
    public List<Integer> transformArray(int[] arr) {
        int length = arr.length;
        boolean flag = true;
        while (flag) {
            flag = false;
            int[] change = new int[length];
            for (int i = 1; i < length - 1; i++) {
                if (arr[i] < arr[i - 1] && arr[i] < arr[i + 1]) {
                    change[i] = 1;
                    flag = true;
                } else if (arr[i] > arr[i - 1] && arr[i] > arr[i + 1]) {
                    change[i] = -1;
                    flag = true;
                }
            }
            if (flag) {
                for (int i = 1; i < length - 1; i++)
                    arr[i] += change[i];
            }
        }
        List<Integer> finalList = new ArrayList<Integer>();
        for (int i = 0; i < length; i++)
            finalList.add(arr[i]);
        return finalList;
    }
}

All Problems

All Solutions