Editing 3382 - robot2 (section)

== Rezolvare == 
<syntaxhighlight lang="python" line="1">
# 3382 - robot2
from typing import Tuple, List

def validare_date(cerinta: int, nr_bomboane: int, dimensiuni_bomboane: List[Tuple[int, int]]) -> bool:
    """
    Verifică dacă datele de intrare corespund constrângerilor problemei.
    Returnează True dacă datele sunt valide și False altfel.
    """
    if not (1 <= cerinta <= 2):
        return False
    if not (2 <= nr_bomboane <= 500000):
        return False
    for x, y in dimensiuni_bomboane:
        if not (0 < x < 100 and 0 < y < 100):
            return False
    return True

def rezolvare_problema(cerinta: int, nr_bomboane: int, dimensiuni_bomboane: List[Tuple[int, int]]) -> int:
    """
    Rezolvă problema pe baza datelor de intrare și returnează răspunsul.
    """
    bomboane_distincte = [0] * 10002
    for x, y in dimensiuni_bomboane:
        bomboane_distincte[min(x, y) * 100 + max(x, y)] += 1
    if cerinta == 1:
        nr_bomboane_distincte = sum(1 for i in range(100, 10002) if bomboane_distincte[i] != 0)
        return nr_bomboane_distincte
    else:
        nr_maxim_cutii = float('inf')
        for i in range(1, 10002):
            if bomboane_distincte[i] != 0 and bomboane_distincte[i] < nr_maxim_cutii:
                nr_maxim_cutii = bomboane_distincte[i]
        return nr_maxim_cutii


if __name__ == '__main__':
    with open('robot.in', 'r') as f_in, open('robot.out', 'w') as f_out:
        cerinta = int(f_in.readline().strip())
        nr_bomboane = int(f_in.readline().strip())
        dimensiuni_bomboane = [tuple(map(int, line.strip().split())) for line in f_in.readlines()]
        if not validare_date(cerinta, nr_bomboane, dimensiuni_bomboane):
            print('Datele nu corespund restrictiilor impuse.')
            f_out.write('Datele nu corespund restrictiilor impuse.')
        else:
            print('Datele sunt introduse corect.')
            raspuns = rezolvare_problema(cerinta, nr_bomboane, dimensiuni_bomboane)
            f_out.write(str(raspuns))


</syntaxhighlight>