PAT 1027 打印沙漏 python

1027 打印沙漏(20)（20 分）

本题要求你写个程序把给定的符号打印成沙漏的形状。例如给定17个“*”，要求按下列格式打印

---

---

*

---

---

所谓“沙漏形状”，是指每行输出奇数个符号；各行符号中心对齐；相邻两行符号数差2；符号数先从大到小顺序递减到1，再从小到大顺序递增；首尾符号数相等。

给定任意N个符号，不一定能正好组成一个沙漏。要求打印出的沙漏能用掉尽可能多的符号。

输入格式：

输入在一行给出1个正整数N（<=1000）和一个符号，中间以空格分隔。

输出格式：

首先打印出由给定符号组成的最大的沙漏形状，最后在一行中输出剩下没用掉的符号数。

输入样例：

19 *

输出样例：

*****
 ***
  *
 ***
*****
2

作者: CHEN, Yue
单位: PAT联盟
时间限制: 200ms
内存限制: 64MB
代码长度限制: 16KB

代码：

n = input().split()
pic = [n[1]]  # 图像列表
num = int(n[0]) - 1
i = 3
while num - i * 2 >= 0:
    pic.append(n[1] * i)   # 在图像最后加一行
    pic.insert(0, n[1] * i)   # 在图像最前端加一行
    num -= 2 * i
    i += 2
for i in pic:
    space = ' ' * int((len(pic[0]) - len(i)) / 2)  # 在前面补上空格
    print("{}{}".format(space, i))
print(num)