8.02 Подсчет количества слов/чисел

Первой цифрой числа может быть либо , либо . На , и месте в слове может стоять любая из цифр. Значит друг может составить различных чисел.
Решение кодом через циклы:

Решение через циклы строится на полном переборе всех четырёхзначных комбинаций цифр от 1 до 9. Сначала создаём пустое множество ans, в котором будем хранить уникальные числа, и определяем строку alf = ’123456789’ с доступными цифрами. Для формирования числа используем четыре вложенных цикла, в которых переменные x1, x2, x3, x4 перебирают все цифры из alf и формируют строку w = x1 + x2 + x3 + x4. После формирования числа проверяем, что первая цифра (x1) равна ’1’ или ’2’. Если условие выполняется, добавляем число в множество ans. В конце работы программы выводим len(ans), что и будет ответом.

# Создаем множество для хранения уникальных чисел
ans = set()
# Определяем цифры, доступные для составления числа
alf = ’123456789’

# Перебор цифр первой позиции числа
for x1 in alf:
    # Перебор цифр второй позиции числа
    for x2 in alf:
        # Перебор цифр третьей позиции числа
        for x3 in alf:
            # Перебор цифр четвертой позиции числа
            for x4 in alf:
                # Формируем число как строку из выбранных цифр
                w = x1 + x2 + x3 + x4
                # Проверяем, что первая цифра равна ’1’ или ’2’
                if x1 in ’12’:
                    # Добавляем число в множество уникальных значений
                    ans.add(w)
# Выводим количество уникальных чисел
print(len(ans))

Решение кодом через itertools:

Для более компактного решения используем функцию product из модуля itertools, которая генерирует все возможные комбинации с повторениями. Сначала импортируем функцию, создаём пустое множество ans и определяем строку alf = ’123456789’ для доступных цифр. С помощью product(alf, repeat=4) формируем все четырёхзначные последовательности. Для каждой последовательности проверяем, что первая цифра (x[0]) равна ’1’ или ’2’, и если условие выполняется, добавляем комбинацию в множество ans. В конце выводим len(ans), что соответствует количеству допустимых чисел.

from itertools import product

# Создаем множество для хранения уникальных чисел
ans = set()
# Цифры, доступные для составления числа
alf = ’123456789’

# Перебираем все четырёхзначные комбинации с повторениями
for x in product(alf, repeat=4):
    # Проверяем, что первая цифра равна ’1’ или ’2’
    if x[0] in ’12’:
        # Добавляем число в множество уникальных значений
        ans.add(x)

# Выводим количество уникальных чисел
print(len(ans))