Bright Tutorials Bright Tutorials
CBSE Class 12 Computer Science Question 20 of 26

Data Structures in Python — Question 21

Back to all questions
21
Question

Question 20

Write a program that, depending upon the user's choice, either adds or removes an element from a Stack.

Solution
def add(stack):
    h = input("Enter element: ")
    stack.append(h)

def remove(stack):
    if len(stack) == 0:
        print("No element to delete")
    else:
        print("Deleted element is:", stack.pop())


stack = []
while True:
    print("1. Add element")
    print("2. Delete element")
    print("3. Exit")
    op = int(input("Enter the Choice: "))
    if op == 1:
        add(stack)
    elif op == 2:
        remove(stack)
    elif op == 3:
        print("Exiting program.")
        break
Output
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 11
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 33
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 44
1. Add element
2. Delete element
3. Exit
Enter the Choice: 2
Deleted element is: 44
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 55
1. Add element
2. Delete element
3. Exit
Enter the Choice: 2
Deleted element is: 55
1. Add element
2. Delete element
3. Exit
Enter the Choice: 3
Exiting program.
Answer

def
add
(
stack
):
h
=
input
(
"Enter element: "
)
stack
.
append
(
h
)
def
remove
(
stack
):
if
len
(
stack
)
==
0
:
print
(
"No element to delete"
)
else
:
print
(
"Deleted element is:"
,
stack
.
pop
())
stack
=
[]
while
True
:
print
(
"1. Add element"
)
print
(
"2. Delete element"
)
print
(
"3. Exit"
)
op
=
int
(
input
(
"Enter the Choice: "
))
if
op
==
1
:
add
(
stack
)
elif
op
==
2
:
remove
(
stack
)
elif
op
==
3
:
print
(
"Exiting program."
)
break
Output
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 11
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 33
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 44
1. Add element
2. Delete element
3. Exit
Enter the Choice: 2
Deleted element is: 44
1. Add element
2. Delete element
3. Exit
Enter the Choice: 1
Enter element: 55
1. Add element
2. Delete element
3. Exit
Enter the Choice: 2
Deleted element is: 55
1. Add element
2. Delete element
3. Exit
Enter the Choice: 3
Exiting program.