PAT1017 打印沙漏(20)

题目链接：
https://www.nowcoder.com/pat/6/problem/4053
题目描述：
本题要求你写个程序把给定的符号打印成沙漏的形状。例如给定17个“*”，要求按下列格式打印

*****
 *** 
  *  
 *** 
*****

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

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

输入描述:

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

输出描述：

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

输入例子：

19 *

输出例子：

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

解题思路：这道题目N比较小，可以通过将奇数从1递增并按照沙漏的排列计算输出符号的总数，符合要求则记录，代码如下：

# TODO: 打印沙漏
N, strr = map(str, input().split(' '))
sum = 0
re = []  # 存储符合条件的奇数
for i in range(1, 1000, 2):     # i为奇数递增
    sum += i
    if sum * 2 - 1 > int(N):    # 判断是否超出  注意奇数1只输出一次
        break
    re.append(i)
si = '{:^' + str(re[-1]) + 's}'   # 用于format格式输出  居中且宽度为re[-1]
count = 0   # 用于计算用掉的符号个数
for i in reversed(re):     # 反转输出沙漏上半部
    temp = ''
    count += i
    for j in range(i):
        temp += strr
    print(si.format(temp))
for i in range(1, len(re)):  # 输出沙漏下半部
    temp = ''
    for j in range(re[i]):
        temp += strr
    print(si.format(temp))
resi = int(N) - (count * 2 - 1)
print(resi)