Вычисление элемента последовательности Люка

Функция, представленная ниже, вычисляет n-ый элемент последовательности Люка, используя простой итеративный подход.

def lucas(n):
    if n == 0:
        return 2
    elif n == 1:
        return 1
    else:
        a, b = 2, 1
        for i in range(2, n+1):
            a, b = b, a + b
        return b

# Пример использования
print(lucas(5)) # Выведет 11

В этом примере, функция lucas начинается с первых двух чисел 2 и 1, после чего каждое последующее число является суммой двух предыдущих чисел.