w3resource

Java: Remove a specific element from a stack

Java Stack: Exercise-13 with Solution

Write a Java program to remove a specific element from a stack.

Sample Solution:

Java Code:

import java.util.Scanner;

public class Stack {
  private int[] arr;
  private int top;

  // Constructor to initialize the stack
  public Stack(int size) {
    arr = new int[size];
    top = -1;
  }

  // Method to push an element onto the stack
  public void push(int num) {
    if (top == arr.length - 1) {
      System.out.println("Stack is full");
    } else {
      top++;
      arr[top] = num;
    }
  }

  // Method to pop an element from the stack
  public int pop() {
    if (top == -1) {
      System.out.println("Stack Underflow");
      return -1;
    } else {
      int poppedElement = arr[top];
      top--;
      return poppedElement;
    }
  }

  // Method to get the top element of the stack
  public int peek() {
    if (top == -1) {
      System.out.println("Stack is empty");
      return -1;
    } else {
      return arr[top];
    }
  }

  // Method to check if the stack is empty
  public boolean isEmpty() {
    return top == -1;
  }

  // Method to remove a specific element from the stack
  public void removeElement(int element) {
    int index = -1;
    for (int i = 0; i <= top; i++) {
      if (arr[i] == element) {
        index = i;
        break;
      }
    }

    if (index == -1) {
      System.out.println("Element not found in the stack");
    } else {
      for (int i = index; i < top; i++) {
        arr[i] = arr[i + 1];
      }
      top--;
      System.out.println("Element removed successfully");
    }
  }

  // Method to display the elements of the stack
  public void display() {
    if (top == -1) {
      System.out.println("Stack is empty");
    } else {
      System.out.print("Stack elements: ");
      for (int i = top; i >= 0; i--) {
        System.out.print(arr[i] + " ");
      }
      System.out.println();
    }
  }

  public static void main(String[] args) {
    System.out.println("Initialize a stack:");
    Stack stack = new Stack(5);
    System.out.println("\nInput some elements on the stack:");
    stack.push(1);
    stack.push(2);
    stack.push(3);
    stack.push(4);
    stack.push(5);
    stack.display();
    Scanner scanner = new Scanner(System.in);
    System.out.print("Input an element to remove from the stack: ");
    int element = scanner.nextInt();
    stack.removeElement(element);
    stack.display();
  }
}

Sample Output:

Input some elements on the stack:
Stack elements: 5 4 3 2 1
Input an element to remove from the stack: 3
Element removed successfully
Stack elements: 5 4 2 1

Flowchart:

Flowchart: Java  Exercises: Remove a specific element  from a stack.
Flowchart: Java  Exercises: Remove a specific element  from a stack.
Flowchart: Java  Exercises: Remove a specific element  from a stack.

Live Demo:

Java Code Editor:

Improve this sample solution and post your code through Disqus

Java Stack Previous: Rotate the stack elements to the left.
Java Stack Exercises Next: Swap the top two elements of a stack.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Become a Patron!

Follow us on Facebook and Twitter for latest update.

It will be nice if you may share this link in any developer community or anywhere else, from where other developers may find this content. Thanks.

https://www.w3resource.com/java-exercises/stack/java-stack-exercise-13.php