Python | A simple Stack Class

tahirraza_se

Tahir Raza

Posted on June 12, 2022

Python | A simple Stack Class

Today I want to talk about a famous, basic but pretty wonderful data structure we all read in our universities or colleges.

Stack

A stack is a linear data structure that follows the principle of Last In First Out (LIFO). This means the last element inserted inside the stack is removed first.

You can think of the stack data structure as the pile of plates on top of another.
stack-of-plates

Here, you can:

  • Put a new plate on top
  • Remove the top plate
  • And, if you want the plate at the bottom, you must first remove all the plates on top. This is exactly how the stack data structure works.

Applications

  • To reverse a word - Put all the letters in a stack and pop them out. Because of the LIFO order of stack, you will get the letters in reverse order.
  • In compilers - Compilers use the stack to calculate the value of expressions like 2 + 4 / 5 * (7 - 9).
  • In browsers - The back button in a browser saves all the URLs you have visited previously in a stack.

Key functions of a Stack

  • Push: Add new element
  • Pop: Remove an element from last
  • Peek: Get the last element without removing it

Cherry on Top

In the following implementation instead of using built-in functions to check empty or length of stack we will use an extra variable count to keep track of these things.

Code

from typing import Type

class Stack:

    elements = []

    count = -1

    def push(self, data):
        self.elements.append(data)
        self.count += 1

    def pop(self) -> int:
        if self.isEmpty() is True:
            raise Exception('Nothing to pop!')

        self.count -= 1
        self.elements.pop()

    def peek(self):
        if self.isEmpty() is True:
            raise Exception('Cannot peek an empty stack')

        return self.elements[self.count]

    def isEmpty(self):
        return self.count < 0
Enter fullscreen mode Exit fullscreen mode
💖 💪 🙅 🚩
tahirraza_se
Tahir Raza

Posted on June 12, 2022

Join Our Newsletter. No Spam, Only the good stuff.

Sign up to receive the latest update from our blog.

Related

While Loops
javascript While Loops

November 26, 2024

Exploratory Testing: A Detailed Guide
javascript Exploratory Testing: A Detailed Guide

November 25, 2024

Bookmarker
webdev Bookmarker

November 24, 2024