2322 - Merry Cristmas

Sursa: [1]

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 ⩽ n ⩽ 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

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.

Rezolvare

#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)

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.