1050 - TCIF

Cerința

Avem la dispoziţie patru numere naturale N, A, B, C, precum şi trei cifre c1, c2, c3 distincte două câte două. Să se determine numărul natural minim, strict mai mare decât N, care are exact A cifre c1, B cifre c2, C cifre c3 şi nu conţine alte cifre.

Date de intrare

Fișierul de intrare tcif.in conține pe prima linie, separate prin câte un spaţiu, numerele naturale A B C c1 c2 c3. Pe linia a doua se află numărul natural N.

Date de ieșire

Fișierul de ieșire tcif.out va conține o singură linie pe care va fi scris cel mai mic număr natural strict mai mare decât N care conţine exact A cifre c1, exact B cifre c2 şi exact C cifre c3 şi nu conţine alte cifre.

Restricții și precizări

• N va avea cel puţin o cifră şi cel mult 1000 de cifre.
• Pentru 10% dintre teste, N ≤ 30 000
• Pentru alte 40% dintre teste, N va avea cel mult 14 cifre
• 0 ≤ c1, c2, c3 ≤ 9; c1, c2 şi c3 sunt distincte două câte două
• 1 ≤ A, 1 ≤ B, 1 ≤ C, A+B+C ≤ 1000
• Datele de intrare sunt alese astfel încât va exista o soluţie.

Exemplu 1

Intrare

2 2 2 3 2 4
293187

Iesire

322344

Rezolvare

from itertools import permutations


def generate_number(A, B, C, c1, c2, c3):
    number_str = c1 * A + c2 * B + c3 * C
    for perm in permutations(number_str):
        yield int(''.join(perm))


def main():
    # Citirea datelor de intrare
    with open("tcif.in", "r") as f:
        A, B, C, c1, c2, c3 = f.readline().strip().split()
        A, B, C = int(A), int(B), int(C)
        N = int(f.readline().strip())

    assert 1 <= A <= 1000, "A trebuie să fie între 1 și 1000."
    assert 1 <= B <= 1000, "B trebuie să fie între 1 și 1000."
    assert 1 <= C <= 1000, "C trebuie să fie între 1 și 1000."
    assert len(str(N)) <= 1000, "N trebuie să aibă cel puțin o cifră și cel mult 1000 de cifre."
    assert 0 <= int(c1) <= 9 and 0 <= int(c2) <= 9 and 0 <= int(c3) <= 9, "c1, c2 și c3 trebuie să fie cifre distincte."
    assert c1 != c2 and c1 != c3 and c2 != c3, "c1, c2 și c3 trebuie să fie cifre distincte."

    # Generarea și verificarea numerelor
    found = False
    result = None
    for num in sorted(generate_number(A, B, C, c1, c2, c3)):
        if num > N:
            result = num
            found = True
            break

    # Afișarea rezultatului
    with open("tcif.out", "w") as f:
        if found:
            f.write(f"{result}\n")
        else:
            f.write("Nu exista solutie\n")


if __name__ == "__main__":
    main()