1050 - TCIF
De la Universitas MediaWiki
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()