Editing 4299 - Gravitatie (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
# Funcția de validare verifică dacă datele de intrare sunt în intervalul specificat
def validare(n_validare, h_validare):
    # Verificăm dacă n este în intervalul 1-10000
    if n_validare < 1 or n_validare > 10000:
        raise ValueError  # Ridicăm o eroare dacă n nu este în intervalul 1-10000
    for h in h_validare:    # Parcurgem lista de înălțimi
        # Verificăm dacă înălțimea este în intervalul 0-2^64-1
        if h < 0 or h > 2**64-1:
            raise ValueError
    file_out.write("Datele de intrare corespund restrictiilor impuse\n")


# Funcția durere_minima calculează gradul minim de durere ce se poate obține prin rearanjarea turnulețelor
def durere_minima(n_minim, h_minim):
    # Creăm o listă de perechi (înălțime, poziție inițială)
    turnuri = [(h_minim[i], i) for i in range(n_minim)]
    # Sortăm lista în funcție de înălțime
    turnuri.sort()
    # Calculăm numărul de bucăți în care se rupe placa de plastic
    calculare = 1
    for i in range(1, n_minim):
        if turnuri[i-1][1] > turnuri[i][1] or turnuri[i-1][0] != turnuri[i][0]:
            calculare += 1
    return calculare


if __name__ == '__main__':
    file_in = open("gravitatiein.txt", "r")
    file_out = open("gravitatieout.txt", "w")

    try:
        # Citim numărul de turnulețe
        n_main = int(file_in.readline())
        # Citim înălțimile turnulețelor
        H_main = list(map(int, file_in.readline().split()))
        # Validăm datele de intrare
        validare(n_main, H_main)
        # Calculăm durerea minimă
        durere = durere_minima(n_main, H_main)
        # Scriem durerea minimă în fișierul de ieșire
        file_out.write(str(durere) + '\n')

    # Dacă datele de intrare nu sunt valide, afișăm un mesaj de eroare
    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")
    # Dacă datele de intrare sunt incomplete, afișăm un mesaj de eroare
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")


</syntaxhighlight>