Vizualizare sursă pentru „2322 - Merry Cristmas”

Sursa: [https://www.pbinfo.ro/probleme/2322/merry-cristmas]
== Cerinţa ==
Într-un ținut îndepărtat cu peisaje de poveste pacea domnește peste întreg ținutul deoarece lumea are absolut tot ce îi trebuie.

Moș Crăciun însa dorește să le mai aducă zâmbetul pe buze și de aceea a decis să ducă fiecărei case câte o sticlă de șampanie deoarece deși oamenii de aici au bani, nu au magazine de unde să cumpere. Însă din păcate Moș Crăciun este bătrân și nu poate transporta prea multe sticle așa că decide să distribuie câte o sticla de șampanie la fiecare două case. Pentru simplitate Moș Crăciun a făcut o hartă a caselor în care o casă este notată cu 1 și pământul gol cu 0. Moș Crăciun însă nu prea le are cu programarea și de aceea vă roagă să îl ajutați!

Scrieți un program care primind harta caselor afișează câte sticle de șampanie trebuie să aducă Moș Crăciun.

== Date de intrare ==
Programul va citi un număr natural n de pe prima linie și de pe a doua linie n numere naturale din intervalul [0;1].

== Date de ieșire == 
 Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele introduse corespund cerintelor.", apoi pe un rând nou afișează   numărul s de sticle necesare . În caz contrar, se va afișa pe ecran: "Datele nu corespund cerintelor."

== Restricţii şi precizări ==
* 1 &les; '''n''' &les; 1.000.000
* cele n numere citite vor fi din intervalul [0;1]
* desigur șampania oferită de Moș Crăciun este una fără alcool
* oricare două case sunt prietene și nu există pereche de case care să refuze șampania

== Exemplul 1 ==
; Intrare
: Introduceți numărul de case: 7
: Introduceți 0 sau 1 pentru casa 1: 1
: Introduceți 0 sau 1 pentru casa 2: 0
: Introduceți 0 sau 1 pentru casa 3: 0
: Introduceți 0 sau 1 pentru casa 4: 1
: Introduceți 0 sau 1 pentru casa 5: 0
: Introduceți 0 sau 1 pentru casa 6: 1
: Introduceți 0 sau 1 pentru casa 7: 1
; Ieșire
: Datele introduse corespund cerintelor.
: 6
<br>

== Exemplul 2 ==
; Intrare
: Introduceți numărul de case: 2
: Introduceți 0 sau 1 pentru casa 1: 6
: Introduceți 0 sau 1 pentru casa 2: 7
; Ieșire
: Datele introduse nu corespund cerintelor.
<br>

== Rezolvare == 
<syntaxhighlight lang="python" line>
#2322
def validate_input(n, lst):
    if not(1 <= n <= 1000000):
        print("Datele introduse nu corespund cerintelor.")
        exit()
    for x in lst:
        if not(0 <= x <= 1):
            print("Datele introduse nu corespund cerintelor.")
            exit()

    print("Datele introduse corespund cerintelor.")


def calculate_champagne(n, lst):
    nr = sum(lst)
    return ((nr-1)*nr)//2


if __name__ == "__main__":
    n = int(input("Introduceți numărul de case: "))
    lst = []
    for i in range(n):
        x = int(input(f"Introduceți 0 sau 1 pentru casa {i+1}: "))
        lst.append(x)
    validate_input(n, lst)
    result = calculate_champagne(n, lst)
    print(result)

</syntaxhighlight>

== Explicatie rezolvare ==
Acest program primește de la utilizator un număr n care reprezintă numărul de case, urmat de n numere care indică dacă fiecare casă acceptă sau nu șampania oferită de Moș Crăciun. Dacă o casă acceptă șampania, valoarea introdusă va fi 1, altfel va fi 0. Programul validează inputul și verifică dacă datele introduse sunt conforme cu cerințele. Mai precis, se verifică dacă n este un număr între 1 și 1.000.000, iar toate cele n numere introduse sunt fie 0, fie 1. Dacă inputul nu este conform, programul va afișa un mesaj de eroare și se va încheia. În caz contrar, programul va calcula numărul minim de sticle de șampanie necesare pentru a servi toate casele care acceptă șampania și va afișa rezultatul.

Funcția '''calculate_champagne''' primește numărul de case n și lista de valori de la input lst și calculează numărul minim de sticle de șampanie necesare în conformitate cu cerințele problemei. Funcția '''validate_input''' primește aceleași date de la input n și lst și verifică dacă inputul este conform cu cerințele problemei. În cazul în care inputul nu este conform, funcția va afișa un mesaj de eroare și va încheia programul.

În funcția main, se citește inputul și se validează folosind funcția validate_input. Dacă inputul este conform, programul calculează numărul minim de sticle de șampanie necesare cu ajutorul funcției '''calculate_champagne''' și afișează rezultatul.