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 '''multimi2.in''' 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 '''multimi2.out''' 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 '''multimi2.out''' 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 ==
; multimi2.in
: 1
: 3 4 3 5
; multimi2.out
: 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 ==
; multimi2.in
: 2
: 3 4 3 5
; multimi2.out
: 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).'''
== Rezolvare ==
<syntaxhighlight lang="python" line>
def multimi2():
    # Deschidem fișierul de intrare pentru a citi datele
    with open('multimi2.in', 'r') as fin:
        p = int(fin.readline().strip())
        N, P1, M, P2 = map(int, fin.readline().split())

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

    # Scriem rezultatul în fișierul de ieșire
    with open('multimi2.out', 'w') as fout:
        if rezultat:
            fout.write(' '.join(map(str, rezultat)))
        else:
            fout.write('-1')

if __name__ == "__main__":
    multimi2()

</syntaxhighlight>