1. 程式人生 > >棧---下一個更大元素

棧---下一個更大元素

  1. 題目
    給定一個迴圈陣列(最後一個元素的下一個元素是陣列的第一個元素),輸出每個元素的下一個更大元素。數字 x 的下一個更大的元素是按陣列遍歷順序,這個數字之後的第一個比它更大的數,這意味著你應該迴圈地搜尋它的下一個更大的數。如果不存在,則輸出 -1。
  2. 示例
輸入: [1,2,1]
輸出: [2,-1,2]
解釋: 第一個 1 的下一個更大的數是 2;
數字 2 找不到下一個更大的數;
第二個 1 的下一個最大的數需要迴圈搜尋,結果也是 2。
注意: 輸入陣列的長度不會超過 10000。
  1. 程式碼實現
    資料結構:棧;維護棧頂元素最小的遞增的序列。
    a.常規演算法找下一個最大元素;
    b.nums陣列最後一個元素入棧,遍歷nums陣列,找比棧中元素下一個更大的數;
    c.未找到下一個更大的元素,置為-1.
import java.util.Stack;

public class Solution {
    public int[] nextGreaterElements(int[] nums) {
        int[] array = new int[nums.length];
        Stack<StackElement> stack = new Stack<>();
        //常規找下一個更大的元素
        for (int i=0;i<nums.length-1;i++){
            StackElement stackElement = new StackElement();
            stackElement.val = nums[i];
            stackElement.index = i;
            stack.push(stackElement);
            int next = nums[i+1];
            while (!stack.empty()&&stack.peek().val<next){
                stackElement = stack.pop();
                array[stackElement.index] = next;
            }
        }
        //nums陣列最後一個元素入棧,注:nums為空的情況
       if(nums.length>0){
           StackElement stackElement = new StackElement();
           stackElement.val = nums[nums.length-1];
           stackElement.index = nums.length-1;
           stack.push(stackElement);
       }
       //找棧中元素的下一個更大的數
        for(int i=0;i<nums.length&&!stack.empty();i++){
            while (!stack.empty()&&stack.peek().val<nums[i]){
                StackElement stackElement = stack.pop();
                array[stackElement.index] = nums[i];
            }
        }
        //未找到
        while (!stack.empty()){
            StackElement stackElement = stack.pop();
            array[stackElement.index] = -1;
        }
        return array;
    }
}

class StackElement{
    int val;
    int index;
}