3187 - RATC2

Cerinţa

Se dau două numere n p și o listă de n elemente cu urcările în autobuz a mai multor persoane a căror structura este prenume nume bilet_platit, fiecare intrare fiind plasată pe câte o linie. Câmpurile prenume și nume sunt șiruri de caractere, iar bilet_platit este un număr care poate fi 0 sau 1; 0 dacă persoana respectivă nu a plătit biletul sau 1 dacă a plătit biletul. Pentru fiecare bilet neplătit se va contoriza o penalizare persoanei.

Se cere :

a) Prenumele, numele și numărul de penalizări al persoanei care are număr maxim de penalizări. Dacă există mai multe persoane cu număr maxim de penalizări se cere afișarea persoanei care apare prima în ordine alfabetică. b) Ordonarea listei de persoane descrescător după numărul de penalizări, la număr de penalizări egale, crescător după prenume, la prenume identice, crescător după nume.

Date de intrare

Fișierul de intrare ratc2.in conține pe prima linie numerele n p, iar pe următoarele n linii datele despre fiecare urcare în autobuz separate prin spații.

Date de ieșire

Dacă p = 1 se va rezolva cerința a). În acest caz, fișierul ratc2.out va conține pe prima linie datele cu semnificația din enunț separate printr-un spațiu. Dacă p = 2 se va rezolva cerința b). În acest caz, fișierul ratc2.out va conține pe câte o linie datele cu semnificația din enunț separate printr-un spațiu.

Restricţii şi precizări

1 ≤ n ≤ 100 șirurile de caractere vor avea maxim 50 de caractere Atenție, persoanele care urcă în autobuz se pot repeta!

Exemplul 1

ratc2.in

5 1
Jotaro Kujo 1
Dio Brando 0
Joseph Joestar 1
Dio Brando 1
Andreescu Ionel 0

ratc2.out

Andreescu Ionel 1

Explicație

Dio Brando și Andreescu Ionel au același număr de penalizări dar Andreescu Ionel e primul în ordine alfabetică.

Exemplul 2

ratc2.in

5 2
Jotaro Kujo 1
Dio Brando 0
Joseph Joestar 1
Dio Brando 1
Andreescu Ionel 0

ratc2.out

Andreescu Ionel 1
Dio Brando 1
Joseph Joestar 0
Jotaro Kujo 0

Explicație

Listă a fost ordonată dupa criteriile precizate la cerința b).

Rezolvare

def read_input(filename):
    with open(filename, 'r') as file:
        n, p = map(int, file.readline().split())
        data = [tuple(file.readline().split()) for _ in range(n)]
    return p, data

def calculate_penalties(data):
    penalties = {}
    for entry in data:
        name = entry[0] + " " + entry[1]
        if entry[2] == '0':
            penalties[name] = penalties.get(name, 0) + 1
    return penalties

def find_max_penalties(penalties):
    max_penalties = max(penalties.values())
    max_penalty_names = [name for name, penalty in penalties.items() if penalty == max_penalties]
    max_penalty_names.sort()
    return max_penalty_names[0], max_penalties

def sort_people(data):
    return sorted(data, key=lambda x: (-penalties.get(x[0] + " " + x[1], 0), x[0], x[1]))

def write_output(filename, result):
    with open(filename, 'w') as file:
        for line in result:
            file.write(" ".join(line) + "\n")

def main():
    p, data = read_input('ratc2.in')
    penalties = calculate_penalties(data)
    
    if p == 1:
        max_name, max_penalty = find_max_penalties(penalties)
        write_output('ratc2.out', [(max_name, str(max_penalty))])
    elif p == 2:
        sorted_data = sort_people(data)
        write_output('ratc2.out', sorted_data)

if __name__ == "__main__":
    main()