2973 - Cate 3 Cifre

Cerința

Fiind dat un număr natural N, să se determine cel mai mare număr natural mai mic sau egal cu N, care are proprietatea că se scrie în exact două baze diferite prin exact 3 simboluri identice. 1. Să se scrie numărul determinat 2. Să se scrie cele două baze determinate și valorile simbolurilor respective.

Date de intrare

Fișierul de intrare cate3cifre.in conţine pe prima linie cerința (1 sau 2). Pe linia a doua a fișierului de intrare se află numărul natural N.

Date de ieșire

Fișierul de ieșire cate3cifre.out va conține pe prima linie, dacă cerința este 1, numărul determinat. Dacă cerința este 2, prima și cea de a doua linie a fișierului de ieșire au aceeași structură: pe fiecare linie se vor scrie, separate printr-un spațiu, două numere naturale b c, reprezentând baza și valoarea simbolului cerut din baza respectivă. Cele două baze se vor afișa în ordine crescătoare.

Restricții și precizări

0 < N ≤ 1000000
Pentru rezolvarea corectă a cerinței 1 se acordă 60 de puncte. Pentru cerința 2, se acordă 30 de puncte.
Pentru 50 de puncte N ≤ 10000
Numărul xyz(b) scris în baza b cu simbolurile x, y, z se scrie în baza 10 ca o valoare calculată astfel: x • b2 + y • b + z (unde simbolurile x, y, z se înlocuiesc cu valorile asociate)
Pentru fiecare test există soluție. În concurs s-au acordat 10 puncte din oficiu. Aici se acordă pentru exemplele din enunț.

Exemplu 1

Intrare

cate3cifre.in

1

1000

Ieșire

Datele introduse corespund restricțiilor impuse.

cate3cifre.out

931

Exemplu 2

Intrare

cate3cifre.in

2

1000

Ieșire

Datele introduse corespund restricțiilor impuse.

cate3cifre.out

11 7

30 1

Rezolvare

#2973 - Cate 3 Cifre
def get_base_representation(num, base):
    result = ""
    while num > 0:
        digit = num % base
        result = str(digit) + result
        num //= base
    return result

def find_number_with_property(N):
    for base1 in range(2, 10):
        for base2 in range(base1 + 1, 10):
            for digit in range(10):
                number = digit * base1**2 + digit * base1 + digit
                if number <= N and get_base_representation(number, base2) == str(digit) * 3:
                    return number, base1, base2, digit

with open("cate3cifre.in", "r") as file_in:
    cerinta = int(file_in.readline().strip())
    N = int(file_in.readline().strip())

if cerinta == 1:
    rezultat = find_number_with_property(N)
    with open("cate3cifre.out", "w") as file_out:
        file_out.write(str(rezultat[0]))
else:
    rezultate = find_number_with_property(N)
    with open("cate3cifre.out", "w") as file_out:
        file_out.write(f"{rezultate[1]} {rezultate[3]}\n")
        file_out.write(f"{rezultate[2]} {rezultate[3]}\n")