1380 - Pluton

Enunț

Î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

Exemplul 1

Intrare

plutonin.txt

10

1223

123

666

321

7890

2213

312

655

1000

1322

Ieșire

Datele de intrare corespund restricțiilor impuse

plutonout.txt

6

3

2

1223 2213 1322

Explicație

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ă.

Exemplul 2

Intrare

plutonin.txt

11

1223

123

666

321

7890

2213

312

655

1000

1322

2000000000

Ieșire

Datele de intrare corespund restricțiilor impuse

Datele de intrare NU corespund restricțiilor impuse

Rezolvare

#1380 - Pluton

def validare_date(n, identificari):
    if 0 < n <= 4000:
        print("Datele de intrare corespund restricțiilor impuse")
    else:
        print("Datele de intrare NU corespund restricțiilor impuse")
        exit(0)

    for identificare in identificari:
        if not 0 <= identificare < 2000000000:
            print("Datele de intrare NU corespund restricțiilor impuse")
            exit(0)


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

        validare_date(n, identificari)
        regrupare_soldati(n, identificari)