PAT 打印沙漏

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

*****

  ***

    *

  ***

*****

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

a,b=input().split()
c=int(a)
d=1
e=0
if c==0:
    print(0)
else:
    while 2*d*d-1<=c:
        d+=1
        e+=1
    for i in range(e-1,-1,-1):
        for j in range(e-1-i):
            print(' ',end = '')
        for k in range(2*i+1):
            print('%s'%b,end = '')
        print()
    for i in range(1,e):
        for j in range(e-1-i):
            print(' ',end = '')
        for k in range(2*i+1):
            print('%s'%b,end = '')
        print()
    print(c-2*e*e+1)

这个题能用到的最大的字符数就是两个首项为 1 的奇数序列的前 n 项和再减去 1 ，即 2*n^2-1 ，然后循环找到这个 n ，先打印沙漏的上半部分，再打印沙漏的下半部分即可。