PTA 1002 写出这个数 python

读入一个正整数 n，计算其各位数字之和，用汉语拼音写出和的每一位数字。

输入格式：
每个测试输入包含 1 个测试用例，即给出自然数 n 的值。这里保证 n 小于 10¹⁰⁰
。

输出格式：
在一行内输出 n 的各位数字之和的每一位，拼音数字间有 1 空格，但一行中最后一个拼音数字后没有空格。

输入样例：

1234567890987654321123456789

输出样例：

yi san wu

代码

def get_pinyin(digit):
    pinyin_dict={0:'ling',1:'yi',2:'er',3:'san',4:'si',
                5:'wu',6:'liu',7:'qi',8:'ba',9:'jiu'}
    return pinyin_dict[digit]

# 读入数字 n
n = int(input())

# 计算 n 的各位数字之和
sum = 0
while n > 0:
    sum += n % 10
    n //= 10
    
# 使用栈存储各位数字
stack = []
while sum > 0:
    # 将 sum 的最后一位数字入栈
    stack.append(sum % 10)
    # 去掉 sum 的最后一位
    sum //= 10

# 逆序输出各位数字
while stack:
    # 取出栈顶数字
    digit = stack.pop()
    # 判断是否是最后一个数字
    if stack:
        # 输出数字对应的拼音，带空格
        print(get_pinyin(digit), end=' ')
    else:
        # 输出数字对应的拼音，不带空格
        print(get_pinyin(digit), end='')

运行结果：