2484 - Key

Enunț

Cristi are în sertar n chei vechi; fiecare a costat o anumită sumă exprimată în lei și fiecare a fost făcută pentru a deschide aceeași ușă. Atât cheile cât și ușa au un cod format din 3 litere. Din păcate, unele chei s-au deteriorat și Cristi le-a împărțit în patru categorii:

1. stricate – nicio o literă din codul cheii nu coincide cu litera de pe aceeași poziție din codul ușii, iar pentru a o repara trebuie sa plătească prețul integral al cheii;

2. deteriorate – exact o litera din codul cheii coincide cu litera de pe aceeași poziție din codul ușii, iar pentru a o repara trebuie sa plătească două treimi din prețul cheii;

3. slab deteriorate – exact două litere din codul cheii coincid cu literele de pe aceleași poziții din codul ușii, iar pentru a o repara trebuie sa platească o treime din prețul cheii;

4. bune – codul cheii e identic cu codul ușii, iar cheia nu trebuie reparată;

Cerințe

1) Să se afle câte chei din fiecare categorie are Cristi.

2) Sa se afle cât a plătit Cristi pentru a repara toate cheile.

Date de intrare

Fișierul de intrare key.in conține pe prima linie un număr P, codul ușii COD și numarul n; dacă P=1 se rezolvă doar cerința 1, iar dacă P=2 se rezolva doar cerința 2. Pe următoarele n linii se citește codul cheii și prețul acesteia.

Date de ieșire

Dacă P=1 se afișează numărul de chei stricate, numărul de chei deteriorate, numărul de chei slab deteriorate și numărul de chei bune separate printr-un spațiu, în aceasta ordine.

Dacă P=2 se afișează doar un număr COST care exprima costul total pentru repararea cheilor.

Restricții și precizări

• 1 ≤ n ≤ 100
• P=1 sau P=2
• orice cod e format din 3 litere
• orice preț este divizibil cu 3 și aparține intervalului deschis [0,999];
• la cerința 2, dacă toate cheile sunt bune sau costul total este 0 se afișează doar mesajul GRATIS
• pentru rezolvarea corectă a cerinței 1 se acordă 30 de puncte, iar pentru rezolvarea corectă a cerinței 2 se acordă 70 de puncte

Exemplu1

key.in

1 ABC 4

ABC 15

AXC 9

AFG 3

UYC 30

key.out

0 2 1 1

Explicație

Cristi nu are cheii stricate, dar are două cheii deteriorate, una slab deteriorata și una buna

Exemplu2

key.in

2 ABC 4

ABC 15

AXC 9

AFG 3

UYC 30

key.out

25

Exemplu3

key.in

3 ABC 4

ABC 15

AXC 9

AFG 3

UYC 30

key.out

Numar invalid: Valoarea lui P trebuie să fie '1' sau '2'.

Rezolvare

#2484 key
with open('key.in', 'r') as f:
    p, door_code, n = f.readline().split()
    n = int(n)
    keys = []
    for i in range(n):
        key_code, price = f.readline().split()
        keys.append((key_code, int(price)))

stricate = deteriorate = slab_deteriorate = bune = 0
total_cost = 0

def validate_input(p, door_code, n, keys):
    if p not in ['1', '2']:
        return False, "Valoarea lui P trebuie să fie '1' sau '2'."

    if not door_code.isalpha() or len(door_code) != 3:
        return False, "Codul ușii trebuie să conțină exact 3 litere."

    if not (1 <= n <= 1000):
        return False, "Numărul de chei (n) trebuie să fie între 1 și 1000."

    if len(keys) != n:
        return False, "Numărul de chei specificat nu corespunde cu numărul real de chei."

    for key_code, price in keys:
        if not key_code.isalpha() or len(key_code) != 3 or not key_code.isupper():
            return False, "Codul cheii trebuie să conțină exact 3 litere majuscule."
        
        try:
            price = int(price)
            if not (1 <= price <= 10000):
                return False, "Prețul cheii trebuie să fie între 1 și 10000."
        except ValueError:
            return False, "Prețul cheii trebuie să fie un număr întreg."

    return True, None

def main():
    with open('key.in', 'r') as f, open('key.out', 'w', encoding='utf-8') as output_file:
        try:
            p, door_code, n = f.readline().split()
            n = int(n)
            keys = [f.readline().split() for _ in range(n)]
        except ValueError:
            output_file.write("Date de intrare invalide.")
            return

        is_valid, error_message = validate_input(p, door_code, n, keys)

        if not is_valid:
            output_file.write("Numar invalid: " + error_message)
            return

        keys = [(key_code, int(price)) for key_code, price in keys]

        stricate = deteriorate = slab_deteriorate = bune = 0
        total_cost = 0

        for key_code, price in keys:
            num_matches = sum(key_code[i] == door_code[i] for i in range(3))
            if num_matches == 0:
                stricate += 1
                total_cost += price
            elif num_matches == 1:
                deteriorate += 1
                total_cost += price * 2 / 3
            elif num_matches == 2:
                slab_deteriorate += 1
                total_cost += price / 3
            else:
                bune += 1

        if p == '1':
            output_file.write(f"{stricate} {deteriorate} {slab_deteriorate} {bune}")
        else:
            if bune == n or total_cost == 0:
                output_file.write("GRATIS")
            else:
                output_file.write(str(int(total_cost)))

if __name__ == "__main__":
    main()