fix: peek() method in stack.ts if statement

code-differently · Oct 23, 2024 · 9937d62 · 9937d62
1 parent 6175f1b
commit 9937d62
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/lesson_12/structs_ts/src/stack.ts b/lesson_12/structs_ts/src/stack.ts
@@ -24,14 +24,13 @@ export class Stack {
   }
 
   peek(): number | null {
-    if (this.isEmpty()) {
-      return null;
+    if (this.top === undefined) {
+      throw new Error('Stack is empty');
     } 
-    const currentTop = this.top;
-    return currentTop ? currentTop.val : null;
+    return this.top.val;
   }
 
   isEmpty(): boolean {
-    return this.top === null;
+    return this.top === undefined;
   }
 }