棧---下一個更大元素
阿新 • • 發佈:2018-12-21
- 題目
給定一個迴圈陣列(最後一個元素的下一個元素是陣列的第一個元素),輸出每個元素的下一個更大元素。數字 x 的下一個更大的元素是按陣列遍歷順序,這個數字之後的第一個比它更大的數,這意味著你應該迴圈地搜尋它的下一個更大的數。如果不存在,則輸出 -1。 - 示例
輸入: [1,2,1]
輸出: [2,-1,2]
解釋: 第一個 1 的下一個更大的數是 2;
數字 2 找不到下一個更大的數;
第二個 1 的下一個最大的數需要迴圈搜尋,結果也是 2。
注意: 輸入陣列的長度不會超過 10000。
- 程式碼實現
資料結構:棧;維護棧頂元素最小的遞增的序列。
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; }