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

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