Fix unhandled EmptyStackException (TheAlgorithms#2606)

aitorfi · web-flow · commit 8094798c4687 · 2021-10-14T13:25:38.000+03:00
diff --git a/Others/QueueUsingTwoStacks.java b/Others/QueueUsingTwoStacks.java
@@ -15,14 +15,14 @@
 class QueueWithStack {
 
   // Stack to keep track of elements inserted into the queue
-  private Stack inStack;
+  private Stack<Object> inStack;
   // Stack to keep track of elements to be removed next in queue
-  private Stack outStack;
+  private Stack<Object> outStack;
 
   /** Constructor */
   public QueueWithStack() {
-    this.inStack = new Stack();
-    this.outStack = new Stack();
+    this.inStack = new Stack<>();
+    this.outStack = new Stack<>();
   }
 
   /**
@@ -82,6 +82,24 @@ public Object peekBack() {
   public boolean isEmpty() {
     return (this.inStack.isEmpty() && this.outStack.isEmpty());
   }
+
+  /**
+   * Returns true if the inStack is empty.
+   * 
+   * @return true if the inStack is empty.
+   */
+  public boolean isInStackEmpty() {
+    return (inStack.size() == 0);
+  }
+
+  /**
+   * Returns true if the outStack is empty.
+   * 
+   * @return true if the outStack is empty.
+   */
+  public boolean isOutStackEmpty() {
+    return (outStack.size() == 0);
+  }
 }
 
 /**
@@ -118,7 +136,7 @@ public static void main(String args[]) {
     System.out.println(myQueue.isEmpty()); // Will print false
 
     System.out.println(myQueue.remove()); // Will print 1
-    System.out.println(myQueue.peekBack()); // Will print NULL
+    System.out.println((myQueue.isInStackEmpty()) ? "null" : myQueue.peekBack()); // Will print NULL
     // instack: []
     // outStack: [(top) 2, 3, 4]
 
