PAT 1027 打印沙漏 (20分)（python 3)

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

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

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

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

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

代码

N,f=input().split()
row=1
while (2*row*row-1)<=int(N):
    row+=1
row-=1
num=2*row-1
a=num
while num>=0:
    # print(a)
    print((str(f)*num).rjust(a))
    a-=1
    num-=2
a+=2
for i in range(2,row+1):
    num=2*i-1
    print((str(f)*num).rjust(a))
    a+=1
sum=2*row**2-1
print(int(N)-sum)

tips：
右侧不用打印空格，否则的话会判定格式错误