Editing 0877 - Cuburi2 (section)

==Rezolvare==

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

def verificare_restrictii(nr_n):    # functia de verificare a datelor de intrare
    if 1 <= nr_n <= 1000:
        return True
    else:
        return False


def procesare_cuburi(nr_n, cuburi):
    # Initializam stiva si lista de numere
    stack = []
    numbers = []

    # Parcurgem fiecare cub
    for i in range(nr_n):
        # Cat timp stiva nu este goala si cubul curent are latura mai mare decat cubul din varful stivei
        while stack and stack[-1] < cuburi[i]:
            # Eliminam cubul din varful stivei
            stack.pop()
            # Eliminam numarul corespunzator cubului eliminat
            numbers.pop()
        # Adaugam cubul curent in stiva
        stack.append(cubes[i])
        # Adaugam numarul corespunzator cubului curent in lista de numere
        numbers.append(i + 1)

    # Afisam numarul de cuburi ramase in stiva
    print(len(numbers))
    # Afisam numerele corespunzatoare cuburilor ramase in stiva
    print(' '.join(map(str, numbers)))


if __name__ == '__main__':
    try:
        n = int(input("Introduceti numarul de cuburi: "))     # citirea numarului de cuburi
        cubes = list(map(int, input().split()))               # citirea lungimilor laturilor cuburilor

        if verificare_restrictii(n):                          # verificam datele de intrare
            print("Datele de intrare corespund restrictiilor impuse.")
            procesare_cuburi(n, cubes)                        # apelam functia de rezolvare
        else:
            print("Datele de intrare nu corespund restrictiilor impuse.")
    # ne asteptam la 2 tipuri de erori din cauza datelor de intrare, le tratam corespunzator
    except ValueError:
        print("Datele de intrare nu corespund restrictiilor impuse.")
    except IndexError:
        print("Datele de intrare nu corespund restrictiilor impuse.")

</syntaxhighlight>