Question

Given a circular array (the next element of the last element is the first element of the array), print the Next Greater Number for every element. The Next Greater Number of a number x is the first greater number to its traversing-order next in the array, which means you could search circularly to find its next greater number. If it doesn't exist, output -1 for this number.

Example 1:

Input: [1,2,1]
Output: [2,-1,2]
Explanation: The first 1's next greater number is 2; 
The number 2 can't find next greater number;
The second 1's next greater number needs to search circularly, which is also 2.

Note: The length of given array won't exceed 10000.

Solution


Approach #1 Brute Force (using Double Length Array) [Time Limit Exceeded]

In this method, we make use of an array which is formed by concatenating two copies of the given array one after the other. Now, when we need to find out the next greater element for , we can simply scan all the elements , such that . The first element found satisfying the given condition is the required result for . If no such element is found, we put a at the appropriate position in the array.

Java

 public class Solution {

    public int[] nextGreaterElements(int[] nums) {
        int[] res = new int[nums.length];
        int[] doublenums = new int[nums.length * 2];
        System.arraycopy(nums, 0, doublenums, 0, nums.length);
        System.arraycopy(nums, 0, doublenums, nums.length, nums.length);
        for (int i = 0; i < nums.length; i++) {
            res[i]=-1;
            for (int j = i + 1; j < doublenums.length; j++) {
                if (doublenums[j] > doublenums[i]) {
                    res[i] = doublenums[j];
                    break;
                }
            }
        }
        return res;
    }
}

Complexity Analysis

  • Time complexity : . The complete array(of size ) is scanned for all the elements of in the worst case.

  • Space complexity : . array of size is used. array of size is used.


Approach #2 Better Brute Force [Accepted]

Instead of making a double length copy of array , we can traverse circularly in the array by making use of the operator. For every element , we start searching in the array(of length ) from the index and look at the next(cicularly) elements. For we do so by scanning over , such that , and we look for the first greater element found. If no such element is found, we put a at the appropriate position in the array.

Java

 public class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int[] res = new int[nums.length];
        for (int i = 0; i < nums.length; i++) {
            res[i] = -1;
            for (int j = 1; j < nums.length; j++) {
                if (nums[(i + j) % nums.length] > nums[i]) {
                    res[i] = nums[(i + j) % nums.length];
                    break;
                }
            }
        }
        return res;
    }
}

Complexity Analysis

  • Time complexity : . The complete array of size is scanned for all the elements of in the worst case.

  • Space complexity : . array of size is used.


Approach #3 Using Stack [Accepted]

This approach makes use of a stack. This stack stores the indices of the appropriate elements from array. The top of the stack refers to the index of the Next Greater Element found so far. We store the indices instead of the elements since there could be duplicates in the array. The description of the method will make the above statement clearer.

We start traversing the array from right towards the left. For an element encountered, we pop all the elements from the stack such that . We continue the popping till we encounter a satisfying . Now, it is obvious that the current only can act as the Next Greater Element for (right now, considering only the elements lying to the right of ).

If no element remains on the top of the stack, it means no larger element than exists to its right. Along with this, we also push the index of the element just encountered(), i.e. over the top of the stack, so that (or ) now acts as the Next Greater Element for the elements lying to its left.

We go through two such passes over the complete array. This is done so as to complete a circular traversal over the array. The first pass could make some wrong entries in the array since it considers only the elements lying to the right of , without a circular traversal. But, these entries are corrected in the second pass.

Further, to ensure the correctness of the method, let's look at the following cases.

Assume that is the correct Next Greater Element for , such that . Now, whenever we encounter , if , it would have already popped the previous and would have become the topmost element. On the other hand, if , it would have become the topmost element by being pushed above the previous . In both the cases, if , it will be correctly determined to be the Next Greater Element.

The following example makes the procedure clear:

!?!../Documents/503_Next_Greater2.json:1000,563!?!

As the animation above depicts, after the first pass, there are a number of wrong entries(marked as ) in the array, because only the elements lying to the corresponding right(non-circular) have been considered till now. But, after the second pass, the correct values are substituted.

Java

public class Solution {

    public int[] nextGreaterElements(int[] nums) {
        int[] res = new int[nums.length];
        Stack<Integer> stack = new Stack<>();
        for (int i = 2 * nums.length - 1; i >= 0; --i) {
            while (!stack.empty() && nums[stack.peek()] <= nums[i % nums.length]) {
                stack.pop();
            }
            res[i % nums.length] = stack.empty() ? -1 : nums[stack.peek()];
            stack.push(i % nums.length);
        }
        return res;
    }
}

Complexity Analysis

  • Time complexity : . Only two traversals of the array are done. Further, atmost elements are pushed and popped from the stack.

  • Space complexity : . A stack of size is used. array of size is used.


Analysis written by: @vinod23

Average Rating: 4.62 (13 votes)