L1-002 打印沙漏 python

python代码，测试已通过

sr=input()
num=int(sr.split(' ')[0])
symbol=sr.split(' ')[1]

whole_num=1
temp_num=1
layer=1
while whole_num<=num:
    temp_num+=2
    whole_num+=temp_num*2
    layer += 2
    if whole_num>num:
        whole_num -= temp_num * 2
        layer -= 2
        break

item = 1
for i in range(layer):
    if i < layer/2:
        print(' '*i, end='')
        print(symbol*(layer-i*2), end='')
        print()
    else:
        item += 2
        print(' ' * (layer-i-1), end='')
        print(symbol * item, end='')
        print()
print(num-whole_num)