Вычисление n-го числа Трибоначчи

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

# Пример вызова функции
print(tribonacci(10)) # Выводит 149