Editing 4301 - Gustare (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
# Funcția de validare verifică dacă datele de intrare sunt în intervalul specificat
def validare(n_validare, m_validare, t_validare, coceni_validare, mere_validare, intrebari_validare):
    if n_validare < 1 or n_validare > 500000:
        raise ValueError
    if m_validare < 1 or m_validare > 500000:
        raise ValueError
    if t_validare < 1 or t_validare > 100:
        raise ValueError
    for valoare in coceni_validare + mere_validare:
        if valoare < 0 or valoare > 1000000000:
            raise ValueError
    for x in intrebari_validare:
        if x < 0 or x > 2000000000:
            raise ValueError
    file_out.write("Datele de intrare corespund restrictiilor impuse.\n")


# Funcția exista_masa verifică dacă există o masă cu valoare nutritivă X
def exista_masa(coceni, mere, intrebari):
    raspunsuri_creare = []
    for x in intrebari:
        exista = 'NU'
        for cocean in coceni:
            if x - cocean in mere:
                exista = 'DA'
                break
        raspunsuri_creare.append(exista)
    return raspunsuri_creare


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

    try:
        # Citim numărul de coceni, numărul de mere și numărul de întrebări
        n_main = int(file_in.readline())
        coceni_main = list(map(int, file_in.readline().split()))
        m_main = int(file_in.readline())
        mere_main = list(map(int, file_in.readline().split()))
        t_main = int(file_in.readline())
        intrebari_main = list(map(int, file_in.readline().split()))
        # Validăm datele de intrare
        validare(n_main, m_main, t_main, coceni_main, mere_main, intrebari_main)
        # Verificăm dacă există o masă cu valoare nutritivă X pentru fiecare întrebare
        raspunsuri = exista_masa(coceni_main, mere_main, intrebari_main)
        # Scriem răspunsurile în fișierul de ieșire
        for raspuns in raspunsuri:
            file_out.write(raspuns + '\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>