PTA 乘法口诀数列

问题：

本题要求你从任意给定的两个 1 位数字 a1​ 和 a2​ 开始，用乘法口诀生成一个数列 {an​}，规则为从 a1​ 开始顺次进行，每次将当前数字与后面一个数字相乘，将结果贴在数列末尾。如果结果不是 1 位数，则其每一位都应成为数列的一项。

输入格式：

输入在一行中给出 3 个整数，依次为 a1​、a2​ 和 n，满足 0≤a1​,a2​≤9，0<n≤103。

输出格式：

在一行中输出数列的前 n 项。数字间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

2 3 10

输出样例：

2 3 6 1 8 6 8 4 8 4

样例解释：

数列前 2 项为 2 和 3。从 2 开始，因为 2×3=6，所以第 3 项是 6。因为 3×6=18，所以第 4、5 项分别是 1、8。依次类推…… 最后因为第 6 项有 6×8=48，对应第 10、11 项应该是 4、8。而因为只要求输出前 10 项，所以在输出 4 后结束。

代码实现：

a1, a2, n = map(int, input().split())
a = [0] * 1000000 
a[0], a[1] = a1, a2
cnt = 2

for i in range(n):
    f = a[i] * a[i + 1]
    if f >= 10:
        a[cnt] = f // 10
        cnt += 1
        a[cnt] = f % 10
        cnt += 1
    else:
        a[cnt] = f
        cnt += 1

print(a[0], end='')
for i in range(1, n):
    print(" " + str(a[i]), end='')

思路：

        首先我们创建一个名为a的数组，长度为1000000，并将其所有元素初始化为0。接着，将a1赋值给a的第一个元素，将a2赋值给a的第二个元素。

        然后，我们使用一个循环来执行以下操作：循环n次，每次都进行一次计算。在每次循环中，我们将当前元素a[i]与下一个元素a[i+1]相乘，得到结果f。

        接下来，我们检查f是否大于等于10。如果是的话，我们需要将f的十位数和个位数分别存储到数组a中。我们通过使用整数除法符号//和取余符号%来实现这一点。然后，我们更新数组a的长度cnt，使其加2，表示我们已经在数组a中添加了两个新的元素。如果f小于10，我们只需要将f存储到数组a中，并更新cnt的值。最后，循环结束后，打印数组a的前n个元素。

（需要注意的是，这段代码存在潜在的性能问题，因为它预先分配了一个长度为1000000的序列，但可能并不需要这么长的序列来存储给定的问题的解。）