PTA 1106 2019数列(Python3)

把 2019 各个数位上的数字 2、0、1、9 作为一个数列的前 4 项，用它们去构造一个无穷数列，其中第 n（>4）项是它前 4 项之和的个位数字。例如第 5 项为 2， 因为 2+0+1+9=12，个位数是 2。

本题就请你编写程序，列出这个序列的前 n 项。

输入格式：

输入给出正整数 n（≤1000）。

输出格式：

在一行中输出数列的前 n 项，数字间不要有空格。

输入样例：

10

输出样例：

2019224758

提交结果：

代码： 

num = int(input())
l = [2, 0, 1, 9]
count = 4
if num <= 4:
    a = l[:num]
    for i in a:
        print(i, end="")
else:
    while len(l) < num:
        a = l[count - 4:count]
        sum1 = sum(a)
        l.append(sum1 % 10)
        count += 1
    for i in l:
        print(i, end="")