[Python] L1-002. 打印沙漏-PAT团体程序设计天梯赛GPLT

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

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

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

输入格式：

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

输出格式：

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

输入样例：
19 *
输出样例：
*****
***
*
***
*****
2

PS：感谢github用户@zhuzihao-hz提供的pull request～

import math

N, a = input().split()
N = int(N)
k = int(math.sqrt((N + 1) / 2))
for i in range(2 * k - 1):
    if i < k:
        for j in range(i):
            print(' ', end='')
        for m in range(2 * (k - i) - 1):
            print(a, end='')
        print('')
    else:
        for j in range(2 * k - i - 2):
            print(' ', end='')
        for m in range(2 * (i - k + 1) + 1):
            print(a, end='')
        print('')
print(N - 2 * k ** 2 + 1)