PAT 乙级（Python） #1027 打印沙漏 (20 分)

#1027 打印沙漏 (20 分)

代码展示:

K, sign = input().split()
K = eval(K)

depth = int(pow((K+1)/2,0.5))
residual = K - (2*depth*depth-1)
delta = -2
num_sign = depth * 2 - 1
for i in range(2*depth-1):
    num_space = ((depth * 2 - 1) - num_sign)//2
    info = (" "*num_space) + (num_sign*sign)
    print(info)
    if num_sign + delta < 0:
        delta = 2
    num_sign += delta

print(residual)

测评结果: