Vizualizare sursă pentru „3203 - SimonaH”

== Cerinta ==

Din perfecţiunea Simonei H. a apărut şi noţiunea de '''p'''-număr, un număr natural cu cifre nenule, ale cărui cifre le putem permuta. Să se afle suma resturilor împărţirii tuturor numerelor obţinute prin permutarea cifrelor lui '''n''' la un număr dat '''p'''.

== Date de intrare ==

Fișierul de intrare '''simonahin.txt''' conține pe prima linie numerele '''n''' şi '''p'''.

== Date de iesire ==

Fișierul de ieșire '''simonahout.txt''' va conține pe prima linie suma resturilor împărţirii tuturor numerelor obţinute prin permutarea cifrelor lui '''n''' la numărul dat '''p'''.

== Restrictii si precizari ==

*1 &les; n &les; 10^18
*2 &les; p &les; 5

== Exemplul 1 ==
;simonahin.txt
:325 5
;simonahout.txt
:Datele introduse corespund restrictiilor impuse
:10

== Exemplul 2 ==
;simonahin.txt
:652 1

:Datele introduse nu corespund restrictiilor impuse


== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
def este_input_valid(n, p):
    return 1 <= n <= 10**18 and 2 <= p <= 5

def putere_modulo(x, y, p):
    result = 1
    x = x % p
    
    while y > 0:
        if y % 2 == 1:
            result = (result * x) % p
        y = y // 2
        x = (x * x) % p
        
    return result

def suma_resturi_permutari(n):
    suma_totala = 0
    
    for p in [2, 3, 4, 5]:
        cifre = [int(digit) for digit in str(n)]
        factorial_inv = 1
        cifre_freq = [0] * 10
        
        for cifra in cifre:
            cifre_freq[cifra] += 1
            
        for i in range(1, 10):
            factorial_inv = (factorial_inv * putere_modulo(i, cifre_freq[i], p)) % p
        
        invers_mod_p = putere_modulo(factorial_inv, p-2, p)
        suma_totala = (suma_totala + invers_mod_p) % p
    
    return suma_totala

if __name__ == "__main__":
    # Citire date de intrare
    n, p = map(int, input("Introduceți numerele n și p, separate prin spațiu: ").split())

    # Verificare validitate date de intrare
    if not este_input_valid(n, p):
        print("Datele introduse nu corespund restricțiilor impuse.")
    else:
        # Calcul și afișare rezultat în fișierul de ieșire
        rezultat = suma_resturi_permutari(n)
        with open("simonahout.txt", "w") as f:
            f.write(str(rezultat) + "\n")


</syntaxhighlight>

== Explicatie ==

Numerele 325, 235, 532, 352, 253, 523 împărţite la 5 dau resturile 0,0,2,2,3,3, suma acestora fiind 10.