Вопрос Python:a. две цифры

На вход подаются две цифры от 0 до 9. Сколько различных положительных чисел можно составить из них, взяв каждую исходную цифру не более одного раза.

В примере это числа: 1, 2, 12, 21.



Sample Input:



1 2

Sample Output:



4

ну да

вспоминайте школьную комбинаторику...

Код: def count_unique_numbers(digit1, digit2):
# Создаем множество для хранения уникальных чисел
unique_numbers = set()

# Добавляем однозначные числа
unique_numbers.add(digit1)
unique_numbers.add(digit2)

# Если цифры различаются, добавляем двузначные числа
if digit1 != digit2:
unique_numbers.add(10 * digit1 + digit2)
unique_numbers.add(10 * digit2 + digit1)

return len(unique_numbers)

# Ввод данных
digit1, digit2 = map(int, input().split())

# Вычисление и вывод результата
result = count_unique_numbers(digit1, digit2)
print(result)