2-fibonacci_last_digit.py

def fib(n):
    a = 0
    b = 1
    for i in range(2, n+1):
        c = a+b
        c = c%10
        a, b = b, c
    return b

n = int(input())
if n <= 1:
    print(n)
    exit()
print(fib(n))