0269 - Puteri K

Cerinţa

Se dau n numere naturale şi un număr natural k. Afişaţi în ordine crescătoare acele numere date care sunt puteri ale lui k.

Date de intrare

Fişierul de intrare puterikin.txt conţine pe prima linie numerele n şi k, iar pe a doua linie n numere naturale separate prin spaţii.

Date de ieşire

Fişierul de ieşire puterikout.txt va conţine pe prima linie numerele care sunt puteri ale lui k, separate printr-un spaţiu, în ordine crescătoare.

Restricții și precizări

• 1 ⩽ n ⩽ 100
• 2 ⩽ k ⩽ 10
• numerele de pe a doua linie a fişierului de intrare vor avea cel mult 8 cifre

Exemplul 1

puterikin.txt

8 2
32 56 317 809 256 2 1 60

puterikout.txt

Datele de intrare corespund restrictiilor impuse
1 2 32 256

Exemplul 2

puterikin.txt

gjkfjgkfjgkfgf

puterikout.txt

Datele de intrare nu corespund restrictiilor impuse

Rezolvare

# Funcția de validare verifică dacă datele de intrare sunt în intervalul specificat
def validare(n_validare, k_validare, numere_validare):
    # Verificăm dacă n este în intervalul 1-100
    if n_validare < 1 or n_validare > 100:
        raise ValueError  # Ridicăm o eroare dacă n nu este în intervalul 1-100
    # Verificăm dacă k este în intervalul 2-10
    if k_validare < 2 or k_validare > 10:
        raise ValueError  # Ridicăm o eroare dacă k nu este în intervalul 2-10
    for numar in numere_validare:    # Parcurgem lista de numere
        # Verificăm dacă numărul are cel mult 8 cifre
        if len(str(numar)) > 8:
            raise ValueError
    file_out.write("Datele de intrare corespund restrictiilor impuse\n")


# Funcția puterik calculează numerele care sunt puteri ale lui k
def puterik(k, numere):
    import math  # Importăm modulul math pentru a folosi funcția log
    # Verificăm dacă fiecare număr este o putere a lui k și creăm o listă cu aceste numere
    puteri_k = [num for num in numere if (num != 0 and ((math.log(num, k) % 1) == 0))]
    puteri_k.sort()  # Sortăm lista în ordine crescătoare
    return puteri_k


if __name__ == '__main__':
    file_in = open("puterikin.txt", "r")
    file_out = open("puterikout.txt", "w")

    try:
        # Citim numărul de numere și k
        n_main, k_main = map(int, file_in.readline().split())
        # Citim numerele
        numere_main = list(map(int, file_in.readline().split()))
        # Validăm datele de intrare
        validare(n_main, k_main, numere_main)
        # Calculăm numerele care sunt puteri ale lui k
        puteri_k_main = puterik(k_main, numere_main)
        # Scriem numerele care sunt puteri ale lui k în fișierul de ieșire
        file_out.write(' '.join(map(str, puteri_k_main)) + '\n')

    # Dacă datele de intrare nu sunt valide, afișăm un mesaj de eroare
    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")
    # Dacă datele de intrare sunt incomplete, afișăm un mesaj de eroare
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")