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.
Exemplul 1:
- Intrare
- 4
- 12 89 941 101
- 5 7
- 8 5 3 9
- Ieșire
- Datele de intrare corespund restrictiilor impuse.
- 1907
Exemplul 2:
- Intrare
- ha63f
- Ieșire
- Datele de intrare nu corespund restrictiilor impuse.
Rezolvare
<syntaxhighlight lang="python" line="1" start="1">
- Functie pentru a verifica daca un numar este prim
def is_prime(pret_n):
if pret_n <= 1:
return False
if pret_n <= 3:
return True
if pret_n % 2 == 0 or pret_n % 3 == 0:
return False
i = 5
while i * i <= pret_n:
if pret_n % i == 0 or pret_n % (i + 2) == 0:
return False
i += 6
return True
- Functie pentru a obtine suma cifrelor unui numar
def sum_digits(pret_n):
return sum(int(digit) for digit in str(pret_n))
- Functie pentru a obtine produsul cifrelor unui numar
def product_digits(pret_n):
product = 1
for digit in str(pret_n):
product *= int(digit)
return product
- Functie pentru a obtine profitul maxim
def max_profit(pret_n, coduri, pret_x, pret_p, pret_d):
profit = 0
for code in coduri:
max_price = pret_n
# 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 = pret_x
# Verificam daca noul cod este prim si adaugam la pret daca este cazul
if is_prime(new_code):
price += new_code + pret_p
# Adaugam la pret pentru fiecare cifra prima din cod
for digit in str(new_code):
if is_prime(int(digit)):
price += pret_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
- Functie pentru a verifica daca datele de intrare sunt valide
def verificare_date_intrare(pret_n, coduri, pret_x, pret_p, pret_d):
if 1 <= pret_n <= 100000 and 1 <= pret_x <= 1000000 and 1 <= pret_p <= 1000000 and len(pret_d) == 4:
for code in coduri:
if 10 <= code <= 999999:
continue
else:
return False
return True
else:
return False
if __name__ == "__main__":
try:
print("Introduceți datele de intrare în următoarea ordine, fiecare pe o linie nouă: numărul de coduri, codurile (separate prin spațiu), prețul x și p (separate prin spațiu), prețul d pentru cifrele 2, 3, 5 și 7 (separate prin spațiu).")
n = int(input())
codes = list(map(int, input().split()))
x, p = map(int, input().split())
d = dict(zip([2, 3, 5, 7], map(int, input().split())))
if verificare_date_intrare(n, codes, x, p, d): # verificam datele de intrare
print("Datele de intrare corespund restrictiilor impuse.")
print(max_profit(n, codes, x, p, d)) # apelam functia de rezolvare
else:
print("Datele de intrare nu corespund restrictiilor impuse.")
# ne asteptam la 2 tipuri de erori din cauza datelor de intrare, le tratam corespunzator
except ValueError:
print("Datele de intrare nu corespund restrictiilor impuse.")
except IndexError:
print("Datele de intrare nu corespund restrictiilor impuse.")
</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.