leetcode-algorithms/src/ApplyOperationsToAnArray.java at master · anishLearnsToCode/leetcode-algorithms · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
// https://leetcode.com/problems/apply-operations-to-an-array
// T: O(N)
// S: O(N)

public class ApplyOperationsToAnArray {
    public int[] applyOperations(int[] array) {
        final int[] result = new int[array.length];
        int j = 0;
        for (int i = 0, previous = array[0] ; i < array.length - 1 ; i++) {
            if (previous == 0) {
                previous = array[i + 1];
                continue;
            }

            if (previous == array[i + 1]) {
                result[j++] = 2 * array[i];
                previous = 0;
                continue;
            }

            result[j++] = array[i];
            previous = array[i + 1];
        }

        if (array[array.length - 2] != array[array.length - 1]) {
            result[j] = array[array.length - 1];
        }

        return result;
    }
}