0875 - Stiva

Cerința

Să se scrie un program care gestionează o stivă de numere întregi. Inițial stiva este vidă. Programul va citi de la tastatură o listă de operații, care pot fi:

• push X – adaugă valoarea întreagă X pe stivă;
• pop – elimină elementul din vârful stivei;
• top – afișează elementul din vârful stivei.

Programul va realiza asupra stivei operațiile citite, în ordine. Afișările se fac pe ecran, câte o valoare pe linie.

Date de intrare

Programul citește de la tastatură un număr n, apoi cele n operații,

Date de ieșire

Programul va afișa pe ecran numere rezultat ale operațiilor top, câte unul pe o linie.

Restricții și precizări

• stiva nu va conține la un moment dat mai mult de 1000 elemente
• elementele stivei vor fi cuprinse între -1.000.000 și 1.000.000
• dacă la un moment dat se aplică operația pop și stiva nu conține niciun element, operația nu are efect
• dacă la un moment dat se aplică operația top și stiva nu conține niciun element, operația nu are efect (nu se afișează nimic)

Exemplul 1:

Intrare

12

push 5

push 4

top

push 7

push -5

pop

top

pop

top

push 11

top

pop

Ieșire

Datele de intrare corespund restrictiilor impuse.

4

7

4

11

Exemplul 2:

Intrare

100000

Ieșire

Datele de intrare nu corespund restrictiilor impuse.

Rezolvare

<syntaxhighlight lang="python" line="1" start="1">

class Stack:

   def __init__(self):
       self.stack = []  # Initializam stiva ca lista goala

   def push(self, x):
       if len(self.stack) < 1000 and -1000000 <= x <= 1000000:  # Verificam daca stiva nu este plina si daca elementul este in intervalul specificat
           self.stack.append(x)  # Adaugam elementul la sfarsitul listei

   def pop(self):
       if len(self.stack) > 0:  # Verificam daca stiva nu este goala
           self.stack.pop()  # Eliminam ultimul element din lista

   def top(self):
       if len(self.stack) > 0:  # Verificam daca stiva nu este goala
           print(self.stack[-1])  # Afisam ultimul element din lista

def verificare_restrictii(nr_n): # functia de verificare a datelor de intrare

   if 1 <= nr_n <= 1000:
       return True
   else:
       return False

if __name__ == '__main__':

   try:
       n = int(input("Introduceti numarul de operatii: "))  # Citim numarul de operatii

       if verificare_restrictii(n):  # verificam datele de intrare
           print("Datele de intrare corespund restrictiilor impuse.")
           s = Stack()  # Cream o instanta a clasei Stack
           for _ in range(n):
               operation = input().split()  # Citim operatia
               if operation[0] == 'push':  # Daca operatia este 'push'
                   s.push(int(operation[1]))  # Adaugam numarul in stiva
               elif operation[0] == 'pop':  # Daca operatia este 'pop'
                   s.pop()  # Eliminam numarul din stiva
               elif operation[0] == 'top':  # Daca operatia este 'top'
                   s.top()  # Afisam numarul din varful stivei
       else:
           print("Datele de intrare nu corespund restrictiilor impuse.")
   except ValueError:
       print("Datele de intrare nu corespund restrictiilor impuse.")

</syntaxhighlight>