1380 - Pluton

Enunt

În timpul acţiunii “Furtuna în deşert” din cauza unei furtuni de nisip, n soldaţi s-au rătăcit de plutoanele lor. După trecerea furtunii se pune problema regrupării acestora pe plutoane. Pentru aceasta se folosesc plăcuţele de identificare pe care soldaţii le poartă la gât. Pe aceste plăcuţe sunt scrise numere care pot identifica fiecare soldat şi plutonul din care acesta face parte. Astfel, soldaţii din acelaşi pluton au numărul de identificare format din aceleaşi cifre, dispuse în altă ordine şi numerele de identificare sunt unice. De exemplu, numerele de identificare 78003433, 83043073, 33347008 indică faptul ca cei trei soldaţi care le poartă fac parte din acelaşi pluton.

Cerința

Fiind date cele n numere de pe plăcuţele de identificare, să se regrupeze cei n soldaţi pe plutoane, indicându-se numărul de plutoane găsite (un pluton refăcut trebuie să aibă minimum un soldat), numărul de soldaţi din cel mai numeros pluton, numărul de plutoane care au acest număr maxim de soldaţi precum şi componenţa unui astfel de pluton (cu număr maxim de soldaţi regrupaţi).

Date de intrare

Fişierul de intrare plutonin.txt conţine pe prima linie numărul n de soldaţi recuperaţi, iar pe fiecare dintre următoarele n linii câte un număr de identificare a celor n soldaţi.

Date de ieșire

Fişierul de ieşire plutonout.txt va conţine pe prima linie numărul de plutoane refăcute. Linia a doua va conţine numărul de soldaţi din cel mai numeros pluton refăcut. Linia a treia va conţine numărul de plutoane care au numărul maxim de soldaţi recuperaţi. Linia a patra va conţine componenţa unui astfel de pluton, cu număr maxim de soldaţi recuperaţi, numerele de identificare ale soldaţilor din componenţă fiind scrise unul după altul separate prin câte un spaţiu.

Restricții și precizări

• 0 ⩽ n ⩽ 4.000
• 0 ⩽ numar de identificare ⩽ 1.999.999.999

Exemplu 1

Intrare

plutonin.txt

10

1223

123

666

321

7890

2213

312

655

1000

1322

Ieșire

plutonout.txt

6

3

2

1223 2213 1322

Exemplu 2

Intrare

plutonin.txt

11

1223

123

666

321

7890

2213

312

655

1000

1322

2000000000

Ieșire

Date de intrare invalide: Numărul de identificare trebuie să fie între 0 și 1999999999 inclusiv.

Rezolvare

#1380 - Pluton
def validare_date(n, identificari):
    if not 0 < n <= 4000:
        raise ValueError("Numărul de soldați trebuie să fie între 1 și 4000 inclusiv.")

    for identificare in identificari:
        if not 0 <= identificare < 2000000000:
            raise ValueError("Numărul de identificare trebuie să fie între 0 și 1999999999 inclusiv.")


def regrupare_soldati(n, identificari):
    plutoane = {}
    for identificare in identificari:
        key = tuple(sorted(str(identificare)))
        plutoane.setdefault(key, []).append(identificare)

    numar_plutoane = len(plutoane)
    numar_max_soldati = max(len(soldati) for soldati in plutoane.values())
    plutoane_max_soldati = [soldati for soldati in plutoane.values() if len(soldati) == numar_max_soldati]

    with open("plutonout.txt", "w") as f:
        f.write(str(numar_plutoane) + "\n")
        f.write(str(numar_max_soldati) + "\n")
        f.write(str(len(plutoane_max_soldati)) + "\n")
        f.write(" ".join(map(str, plutoane_max_soldati[0])) + "\n")


if __name__ == "__main__":
    with open("plutonin.txt", "r") as f:
        n = int(f.readline().strip())
        identificari = [int(f.readline().strip()) for _ in range(n)]

    try:
        validare_date(n, identificari)
        regrupare_soldati(n, identificari)
        print("Regrupare efectuată cu succes. Verificați fișierul pluton.out pentru rezultate.")
    except ValueError as e:
        print(f"Date de intrare invalide: {e}")

Explicatii

Au fost recuperaţi soldaţi din 6 plutoane distincte, cei mai mulţi soldaţi recuperaţi dintr-un pluton fiind în număr de 3. Există 2 plutoane cu număr maxim de soldaţi recuperaţi (3), unul dintre ele fiind format din soldaţii cu numerele 1223 2213 1322. De remarcat că şi soluţia 321 312 123 este corectă.