PAT乙级真题-1027 打印沙漏-python3

本来是很简单的，坑就坑在我1和4答案错了。我也不知道为啥，知道发现是因为！打！印！了！右！边！的！空！格！
直接贴代码吧。我这里是计算了等差数列求和，然后分上下三角打印。

total, ch = input().split()
total = int(total)
cnt = 0
while 1 + cnt * 2 * (cnt + 2) <= total:
    cnt += 1
cnt -= 1
if cnt == 0:
    print(ch)
    print(str(total - 1))
    exit(0)
r = total - (1 + cnt * 2 * (cnt + 2))
ma = 2 * cnt + 1
nrow = ma
for i in range(cnt):
    nb = (nrow-ma) // 2
    print(' ' * nb + ch * ma)
    ma -= 2
nb = (nrow - 1) // 2
print(' ' * nb + ch * 1)
ma = 3
for i in range(cnt):
    nb = (nrow - ma) // 2
    print(' ' * nb + ch * ma)
    ma += 2
print(str(r))
exit(0)

说明：

1. exit(0)是因为PAT要求以0退出
2. 提示非0返回的话，优先检查代码本身是不是有错误，比如这次我忘了把total转成int就出过错误