Editing 3685 - Primest Prime Pancakes

==Cerință==

Pentru a câștiga niște bani în plus, Ștefan s-a decis să deschidă un magazin nou de clătite pentru următorul sezon estival, deoarece toată lumea e înnebunită dupa clătite vara. Magazinul va vinde n tipuri de clătite, fiecare dintre ele având un cod anume(două clătite pot avea același cod), în funcție de tipul clătitei.

Pentru a afla prețul unei clătite, Ștefan a găsit un algoritm interesant pentru a decide prețul clătitei. Pentru că numerele prime sunt favoritele sale, prețul va depinde de cât de “primă” este clătita.

Astfel:

Prețul inițial al clătitei e x euro, x fiind dat în datele de intrare.

Dacă codul clătitei e un număr prim, prețul crește cu cod + p euro, unde cod e valoarea codului și p e dat în datele de intrare.

Pentru fiecare cifră primă din cod (2, 3, 5, 7), prețul va crește cu d[i] euro, unde valorile d[i] sunt date în datele de intrare.

Dacă suma cifrelor codului este număr prim, prețul crește cu sc euro, unde sc e suma cifrelor codului.

Dacă produsul cifrelor codului este număr prim, prețul crește cu pd euro, unde pd e produsul cifrelor din cod.

Din păcate, această metodă poate să nu ducă la un profit îndeajuns de mare și acum a venit vremea să îl ajutați pe Ștefan să obțină cel mai mare profit posibil. Pentru a realiza acest obiectiv, puteți să schimbați cel mult o cifră din fiecare cod pentru a maximiza suma prețurilor clătitelor.

Totuși, nu puteți face prima cifră egală cu 0 pentru a obține un număr cu o cifră mai puțin, chiar dacă ar putea fi cea mai profitabilă alegere.

==Date de intrare==

Pe prima linie programul va citi numărul n, reprezentând numărul de clătite disponibile în magazinul lui Ștefan.

Pe cea de-a doua linie, programul va citi n numere, reprezentând codurile clătitelor.

Cea de-a treia linie va conține două valori, x și p, cu semnificația din enunț.

Cea de-a patra linie va conține patru valori, d[2] , d[3], d[5], d[7].

==Date de ieșire==

Programul va afișa pe ecran profitul maxim pe care Ștefan îl poate obține după modificarea convenabilă a codurilor clătitelor.

==Restricții și precizări==

*1 ≤ n ≤ 100 000
*1 ≤ cod[i] ≤ 999 999
*1 ≤ x, p, d[i] ≤ 1 000 000
*Pentru teste în valoare de 30 de puncte, 1 ≤ n ≤ 1000.

==Exemplu:==

;Intrare
:4
:12 89 941 101
:5 7
:8 5 3 9
;Ieșire
:1907

==Rezolvare==

<syntaxhighlight lang="python" line="1" start="1">

# Functie pentru a verifica daca un numar este prim
def is_prime(n):
    if n <= 1:
        return False
    if n <= 3:
        return True
    if n % 2 == 0 or n % 3 == 0:
        return False
    i = 5
    while i * i <= n:
        if n % i == 0 or n % (i + 2) == 0:
            return False
        i += 6
    return True

# Functie pentru a obtine suma cifrelor unui numar
def sum_digits(n):
    return sum(int(digit) for digit in str(n))

# Functie pentru a obtine produsul cifrelor unui numar
def product_digits(n):
    product = 1
    for digit in str(n):
        product *= int(digit)
    return product

# Functie pentru a obtine profitul maxim
def max_profit(n, codes, x, p, d):
    profit = 0
    for code in codes:
        max_price = x
        # Parcurgem toate cifrele si incercam sa le schimbam cu o alta cifra
        for i in range(10):
            for j in range(len(str(code))):
                # Cream un nou cod prin schimbarea unei cifre
                new_code = list(str(code))
                new_code[j] = str(i)
                new_code = int(''.join(new_code))
                price = x
                # Verificam daca noul cod este prim si adaugam la pret daca este cazul
                if is_prime(new_code):
                    price += new_code + p
                # Adaugam la pret pentru fiecare cifra prima din cod
                for digit in str(new_code):
                    if is_prime(int(digit)):
                        price += d[int(digit)]
                # Verificam daca suma cifrelor codului este numar prim si adaugam la pret daca este cazul
                if is_prime(sum_digits(new_code)):
                    price += sum_digits(new_code)
                # Verificam daca produsul cifrelor codului este numar prim si adaugam la pret daca este cazul
                if is_prime(product_digits(new_code)):
                    price += product_digits(new_code)
                # Actualizam pretul maxim pentru acest cod de clatita
                max_price = max(max_price, price)
        # Adaugam la profit pretul maxim obtinut pentru acest cod de clatita
        profit += max_price
    return profit

# Punctul de intrare al programului
if __name__ == "__main__":
    n = 4
    codes = [12, 89, 941, 101]
    x = 5
    p = 7
    d = {2:8, 3:5, 5:3, 7:9}
    
    print(max_profit(n, codes, x, p, d))

</syntaxhighlight>

==Explicație==
Trebuie să schimbăm prima valoare de la 12 la 17 și vom obține un profit de 45.

Este optim să păstrăm al doilea cod la 89 pentru un profit de 118.

Trebuie să schimbăm cea de-a treia valoare de la 941 la 991 și vom obține un profit de 1022.

Trebuie să schimbăm cea de-a patra valoare de la 101 la 701 și vom obține un profit de 722.

Suma profiturilor va deveni 1907.