Two ways to solve primality testing (part 0ab9)

Покажу решение и коротко сравню его с альтернативным подходом.

Задача

Определить, является ли число n простым. Достаточно проверять делители до квадратного корня из n.

Решение

def is_prime(n: int) -> bool:
    if n < 2:
        return False
    for d in range(2, int(n ** 0.5) + 1):
        if n % d == 0:
            return False
    return True


print(is_prime(29))

Как это работает

Если у числа есть делитель, больший корня, то у него есть и парный меньший. Поэтому перебор до √n гарантированно находит любой составной делитель.

Пример работы

True

Сложность

O(√n) по времени, O(1) по памяти.

Альтернативный подход

Ту же задачу часто решают иначе: например, через рекурсию вместо цикла или через стандартную библиотеку вместо ручной реализации. Альтернатива обычно короче, но первое «ручное» решение лучше показывает механику работы алгоритма. Полезно написать оба и сравнить на одинаковых тестах.

Оба подхода рабочие; важно понимать цену каждого, а не копировать первый попавшийся.

Читайте также:

находит · return · рекурсию · одинаковых · работы · короче · проверять · квадратного · тестах · Альтернатива · вместо · корня · рабочие · составной · False · цикла · int(n · коротко · альтернативным · Задача