503. Next Greater Element II

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.

 1 public class Solution {
 2     public int[] nextGreaterElements(int[] A) {
 3         int n = A.length, res[] = new int[n];
 4         Arrays.fill(res, -1);
 5         Stack<Integer> stack = new Stack<>();
 6         for (int i = 0; i < n * 2; i++) {
 7             while (!stack.isEmpty() && A[stack.peek()] < A[i % n])
 8                 res[stack.pop()] = A[i % n];
 9             stack.push(i % n);
10         }
11         return res;
12     }
13 }

 

posted @ 2020-03-05 15:16  北叶青藤  阅读(222)  评论(0)    收藏  举报