Editing 0622 - Multimi 2

== Cerinţa ==
Cunoscând '''N''', '''P1''', '''M''' și '''P2''' se cere:

a) Să se afișeze reuniunea celor două mulțimi.
b) Să se afișeze intersecția celor două mulțimi.
== Date de intrare ==
Fişierul de intrare '''multimi2in.txt''' conţine pe prima linie un număr natural '''p'''. Pentru toate testele de intrare, numărul p poate avea doar valoarea '''1''' sau valoarea '''2'''. Pe a doua linie se află patru numere naturale '''N''', '''P1''', '''M''' și '''P2''' cu semnificația din enunț.
== Date de ieşire ==
Dacă valoarea lui '''p''' este '''1''', '''se va rezolva numai punctul a)''' din cerință. În acest caz, în fişierul de ieşire '''multimi2out.txt''' se va scrie reuniunea celor două mulțimi.

Dacă valoarea lui '''p''' este '''2''', '''se va rezolva numai punctul b)''' din cerință. În acest caz, în fișierul de ieșire '''multimi2out.txt''' se va scrie intersecția celor două mulțimi.
== Restricții și precizări ==
* '''1 ≤ N, M  ≤ 500000'''
* '''1 ≤ P1, P2  ≤ 750000'''
* Pentru rezolvarea corectă a primei cerinţe se acordă '''50''' de puncte, iar pentru cerința a doua se acordă '''50''' de puncte.
* Elementele mulțimii din fișierul de ieșire se vor afişa în ordine crescătoare.
* Dacă mulţimea rezultată este vidă se va afişa '''-1'''.
== Exemplul 1 ==
; multimi2in.txt
: 1
: 3 4 3 5
; multimi2out.txt
: Datele introduse corespund restricțiilor impuse.
: 4 5 6 7
== Explicație ==
'''p = 1'''
Prima mulțime este '''{4,5,6}''', iar a doua '''{5,6,7}'''.
Reuniunea celor două mulțimi reprezintă elementele care aparțin cel puțin unei mulțimi.
'''Atenție! Pentru acest test se rezolvă doar cerința a).'''
== Exemplul 2 ==
; multimi2in.txt
: 2
: 3 4 3 5
; multimi2out.txt
: Datele introduse corespund restricțiilor impuse.
: 5 6
== Explicație ==
'''p = 2'''
Prima mulțime este '''{4,5,6}''', iar a doua '''{5,6,7}'''.
Intersecția celor două mulțimi reprezintă elementele comune.
'''Atenție! Pentru acest test se rezolvă doar cerința b).'''
== Exemplul 3 ==
; multimi2in.txt
: 3
: 5 1 5 1
; multimi2out.txt
: Datele introduse nu corespund restricțiilor impuse.
== Rezolvare ==
<syntaxhighlight lang="python" line>
# 0622 - Multimi 2
def validare(p, n, p1, m, p2):           # functia de validare a datelor de intrare
    if not (1 <= p <= 2) or not (1 <= n, m <= 1000) or not (0 <= p1, p2 <= 10000):
        raise ValueError("Datele de intrare nu corespund restrictiilor impuse")
    else:
        return "Datele de intrare corespund restrictiilor impuse"


def multimi2(p, n, p1, m, p2):                     # functia de rezolvare
    # Generăm cele două mulțimi
    multime1 = set(range(p1, p1 + n))
    multime2 = set(range(p2, p2 + m))

    # Calculăm rezultatele în funcție de cerința specificată
    if p == 1:
        # Cerința a): reuniunea celor două mulțimi
        rezultat = sorted(multime1.union(multime2))
    else:
        # Cerința b): intersecția celor două mulțimi
        rezultat = sorted(multime1.intersection(multime2))

    return rezultat


def main():
    with open('multimi2in.txt', 'r') as fin, open('multimi2out.txt', 'w') as fout:
        try:
            p = int(fin.readline().strip())
            p1, p1, m, p2 = map(int, fin.readline().split())
            validare_message = validare(p, p1, p1, m, p2)                 # apelul functiei de validare
            fout.write(validare_message + "\n")
            rezultat = multimi2(p, p1, p1, m, p2)    # apelul functiei de rezolvare
            if rezultat:
                fout.write(' '.join(map(str, rezultat)))
            else:
                fout.write('-1')
        except ValueError as ve:
            fout.write(str(ve))


if __name__ == "__main__":
    main()

</syntaxhighlight>