Vizualizare sursă pentru „3298 - fmi orase2”

Sursa: [https://www.pbinfo.ro/probleme/3298/fmi-orase2 - fmi_orase2]
----
== Cerinţa ==
Suntem în anul 2050. Resursele de apă de pe planeta noastră sunt limitate din cauza schimbărilor climatice. Pentru ca toți locuitorii unui oraș să supraviețuiască în cazul unei catastrofe, orașul trebuie să aibă o rezervă de cel puțin '''1 km^3''' de apă. Pentru a face față mai ușor unor catastrofe, orașele pot forma alianțe în care se pot împrumuta reciproc cu apă. Astfel, într-o alianță formată din '''h''' orașe, fiecare oraș trebuie să aibă o rezervă de cel puțin '''h km^3''' de apă (pentru a putea supraviețui toți locuitorii săi și a putea să împrumute cu câte '''1 km^3''' de apă fiecare dintre celelalte '''h - 1''' orașe aliate).

Fiind dat numărul '''m''' de orașe și un șir de '''m''' numere naturale strict pozitive, notate cu '''c[1] c[2] ... c[m]''', unde fiecare număr '''c[i]''' reprezintă capacitatea maximă de stocare a orașului '''i''', să se determine numărul maxim de orașe '''h''' care pot forma o alianță, adică există cel puțin '''h''' orașe care pot reține fiecare o cantitate de apă cel puțin egală cu '''h km^3'''.
== Date de intrare ==
Fișierul de intrare '''fmi_orase2.in''' conține pe prima linie numărul '''m''', iar pe a doua linie '''m''' numere naturale separate prin spații.
== Date de ieșire == 
Dacă datele sunt introduse corect, pe ecran se va afișa: 
"Datele sunt corecte.", iar apoi in fișierul de ieșire '''fmi_orase2.out''' va conține pe prima linie numărul '''h''', reprezentând cel mai mare număr de orașe care se pot alia. În caz contrar, se va afișa pe ecran: "Datele nu sunt comform restricțiilor impuse.".

== Restricţii şi precizări ==
* 1 ≤ '''m''' ≤ 100
* numerele de pe a doua linie a fișierului de intrare vor fi mai mici decât '''1.000'''
== Exemple ==
===Exemplul 1===
; fmi_orase2.in
: 5
: 3 1 6 1 5
; Ieșire
: Datele sunt corecte.
; fmi_orase2.out
: 3
===Exemplul 2===
; fmi_orase2.in
: 5
: 7 81 2 3 4
; Ieșire
: Datele sunt corecte.
; fmi_orase2.out
: 4
===Exemplul 3===
; fmi_orase2.in
: 2 
: 314441 41241241
; Ieșire
: Datele nu sunt comform restricțiilor impuse.
<br>
== Rezolvare == 
<syntaxhighlight lang="python" line>
#3298 - fmi orase2

def orase(vector,m):
    f = open("fmi_orase2.out", "w")
    vector.sort(reverse=True)
    h = 0
    for i in range(m):
        h += 1
        if h >= vector[i]:
            break
    f.write(str(h))
    
    
def conform_restrictiilor():
    with open("fmi_orase2.in", "r") as f:
        m =  int(f.readline())
        vector = list(map(int, f.readline().split()))
        if not (1 <= m <= 100):
            print("Datele nu sunt conform restricțiilor impuse.")
            exit()
        for x in vector:
            if x >= 1000:
                print("Datele nu sunt conform restricțiilor impuse.")
                exit()
        print("Datele sunt corecte.")
        return vector, m
        

if __name__ == '__main__':
    vector , m = conform_restrictiilor()
    orase(vector , m)

</syntaxhighlight>

==Explicaţie cod==

Acesta este un cod care citeste date dintr-un fisier de intrare '''fmi_orase2.in''' si afiseaza rezultatul in alt fisier '''fmi_orase2.out'''. Codul verifica daca datele citite sunt in limitele permise si daca sunt corecte.

Functia '''conform_restrictiilor''' citeste datele din fisier si verifica daca sunt in limitele permise. Daca datele sunt in limitele permise, aceasta returneaza o lista cu numerele citite si numarul de orase '''m'''.

Functia '''orase''' sorteaza lista cu numerele citite in ordine descrescatoare, calculeaza suma tuturor oraselor incepand cu orasul cu populatia maxima, si scrie aceasta suma in fisierul de iesire '''fmi_orase2.out'''.

In final, programul principala citeste datele, verifica daca sunt in limitele permise, si daca acestea sunt corecte, calculeaza si scrie rezultatul in fisierul de iesire.