Java程式:查詢棧中最大和最小元素
棧是一種基本的遵循後進先出 (LIFO) 原則的資料結構。棧有很多用例,例如組織函式呼叫和撤銷操作。通常,人們可能會遇到查詢棧中最大和最小元素的問題,本文將演示使用 Java 完成此任務的多種方法。
理解棧
棧是一種線性資料結構,只允許在一端進行操作,稱為棧頂。主要操作包括:
- Push(壓棧):將元素新增到棧頂。
- Pop(出棧):移除並返回棧頂元素。
- Peek(檢視):檢視棧頂元素,但不移除它。
- IsEmpty(是否為空):檢查棧是否為空。
問題陳述
目標是確定棧中的最大和最小元素。考慮到棧的 LIFO 特性,無法直接訪問除棧頂之外的元素。這就需要遍歷棧,同時跟蹤最大值和最小值。
使用兩個附加變數
這裡我們使用兩個變數 min 和 max 分別跟蹤最小值和最大值。遍歷棧,並在處理每個元素時更新這些變數。這是最簡單的方法,也是最耗時和最耗空間的方法。
import java.util.Stack; public class MaxMinInStack { public static void main(String[] args) { Stack < Integer > stack = new Stack < > (); stack.push(10); stack.push(20); stack.push(30); stack.push(5); stack.push(15); int[] result = findMaxMin(stack); System.out.println("Maximum element: " + result[0]); System.out.println("Minimum element: " + result[1]); } public static int[] findMaxMin(Stack <Integer> stack) { if (stack.isEmpty()) { throw new IllegalArgumentException("Stack is empty"); } int max = Integer.MIN_VALUE; int min = Integer.MAX_VALUE; for (Integer element: stack) { if (element > max) { max = element; } if (element < min) { min = element; } } return new int[] { max, min }; } }
輸出
Maximum element: 30 Minimum element: 5
使用輔助棧
這裡我們透過使用 pop 操作遍歷棧並更新 min 和 max。輔助棧臨時儲存元素,然後將其恢復到原始棧中。
import java.util.Stack; public class MaxMinInStack { public static void main(String[] args) { Stack <Integer> stack = new Stack < > (); stack.push(10); stack.push(20); stack.push(30); stack.push(5); stack.push(15); int[] result = findMaxMinWithAuxiliaryStack(stack); System.out.println("Maximum element: " + result[0]); System.out.println("Minimum element: " + result[1]); } public static int[] findMaxMinWithAuxiliaryStack(Stack <Integer> stack) { if (stack.isEmpty()) { throw new IllegalArgumentException("Stack is empty"); } Stack <Integer> tempStack = new Stack < > (); int max = stack.peek(); int min = stack.peek(); while (!stack.isEmpty()) { int current = stack.pop(); if (current > max) { max = current; } if (current < min) { min = current; } tempStack.push(current); } // Restore the original stack while (!tempStack.isEmpty()) { stack.push(tempStack.pop()); } return new int[] { max, min }; } }
輸出
Maximum element: 30 Minimum element: 5
使用兩個棧
這種方法使用兩個額外的棧,一個用於記住最大元素 (maxStack),另一個用於記住最小元素 (minStack)。每次新元素進入主棧時,如果它使最大值或最小值變大,我們也會將其放入 maxStack 或 minStack 中。
import java.util.Stack; public class MaxMinInStack { public static void main(String[] args) { Stack <Integer> stack = new Stack < > (); stack.push(10); stack.push(20); stack.push(30); stack.push(5); stack.push(15); int[] result = findMaxMinWithTwoStacks(stack); System.out.println("Maximum element: " + result[0]); System.out.println("Minimum element: " + result[1]); } public static int[] findMaxMinWithTwoStacks(Stack <Integer> stack) { Stack <Integer> maxStack = new Stack < > (); Stack <Integer> minStack = new Stack < > (); while (!stack.isEmpty()) { int current = stack.pop(); if (maxStack.isEmpty() || current >= maxStack.peek()) { maxStack.push(current); } if (minStack.isEmpty() || current <= minStack.peek()) { minStack.push(current); } } int max = maxStack.peek(); int min = minStack.peek(); return new int[] { max, min }; } }
輸出
Maximum element: 30 Minimum element: 5
使用修改後的棧結構
修改棧結構,使其本身包含最大值和最小值以及常規棧元素。每個元素都儲存為一個包含值、當前最大值和當前最小值的對。
import java.util.Stack; public class MaxMinInStack { static class StackNode { int value; int currentMax; int currentMin; StackNode(int value, int currentMax, int currentMin) { this.value = value; this.currentMax = currentMax; this.currentMin = currentMin; } } public static void main(String[] args) { Stack <StackNode> stack = new Stack < > (); push(stack, 10); push(stack, 20); push(stack, 30); push(stack, 5); push(stack, 15); int[] result = findMaxMinWithModifiedStack(stack); System.out.println("Maximum element: " + result[0]); System.out.println("Minimum element: " + result[1]); } public static void push(Stack <StackNode> stack, int value) { int max = stack.isEmpty() ? value : Math.max(value, stack.peek().currentMax); int min = stack.isEmpty() ? value : Math.min(value, stack.peek().currentMin); stack.push(new StackNode(value, max, min)); } public static int[] findMaxMinWithModifiedStack(Stack <StackNode> stack) { if (stack.isEmpty()) { throw new IllegalArgumentException("Stack is empty"); } StackNode topNode = stack.peek(); return new int[] { topNode.currentMax, topNode.currentMin }; } }
輸出
Maximum element: 30 Minimum element: 5
結論
查詢棧中最大和最小元素可以使用不同的方法,每種方法都有其優點和權衡。所示方法包括使用額外變數、輔助棧、為最大值和最小值管理單獨的棧或更改棧本身的結構。
每種技術都提供了一種特定方法來處理訪問或儲存棧元素的問題,這使其適用於某些情況,具體取決於記憶體限制、效能要求和資料完整性需求。瞭解和應用這些方法可以幫助開發人員有效地處理 Java 中的棧,使他們的應用程式最適合某些情況。
廣告