如题目，本人用 Python 解题，求指教

Subtask #1 AC，其余全 TLE 以下是代码：

def fib(x):
    a, b = 1, 1
    if x == 1 or x == 2:
        return 1
    for i in range(x-2):
        a, b = b, a+b
    return b

def s(number):
    sum_of_digits = 0
    while number > 0:
        digit = number % 10
        sum_of_digits += digit
        number //= 10
    return sum_of_digits
    
def main(n):
    ss = 0
    for i in range(1, n+1):
        ss += s(fib(i))
    return ss % 9

T = int(input())
arr = []
for i in range(T):
    arr.append(int(input()))
for i in [main(i) for i in arr]:
    print(i)