Thursday, October 20, 2016

Range Addition

Assume you have an array of length n initialized with all 0's and are given k update operations.
Each operation is represented as a triplet: [startIndex, endIndex, inc] which increments each element of subarray A[startIndex ... endIndex] (startIndex and endIndex inclusive) with inc.
Return the modified array after all k operations were executed.
Example:
Given:

    length = 5,
    updates = [
        [1,  3,  2],
        [2,  4,  3],
        [0,  2, -2]
    ]

Output:

    [-2, 0, 3, 5, 3]
Explanation:
Initial state:
[ 0, 0, 0, 0, 0 ]

After applying operation [1, 3, 2]:
[ 0, 2, 2, 2, 0 ]

After applying operation [2, 4, 3]:
[ 0, 2, 5, 5, 3 ]

After applying operation [0, 2, -2]:
[-2, 0, 3, 5, 3 ]
Hint:
  1. Thinking of using advanced data structures? You are thinking it too complicated.
  2. For each update operation, do you really need to update all elements between i and j?
  3. Update only the first and end element is sufficient.
  4. The optimal time complexity is O(k + n) and uses O(1) extra space.

The idea is to utilize the fact that the array initializes with zero. The hint suggests us that we only needs to modify the first and last element of the range. In fact, we need to increment the first element in the range and decreases the last element + 1 (if it's within the length) by inc. Then we sum up all previous results. Why does this work? When we sum up the array, the increment is passed along to the subsequent elements until the last element. When we decrement the end + 1 index, we offset the increment so no increment is passed along to the next element.

**This only works for array initialized with 0.


public int[] getModifiedArray(int length, int[][] updates) {
        int[] nums = new int[length];
        if (length == 0) {
            return nums;
        }
        for (int[] u : updates) {
            nums[u[0]] += u[2];
            if (u[1] + 1 < length) {
                nums[u[1] + 1] -= u[2];
            }
        }
        for (int i = 1; i < length; i++) {
            nums[i] += nums[i - 1];
        }
        return nums;
    }


No comments:

Post a Comment