PAT (Basic Level) Practice （中文）1027 打印沙漏

题目

if __name__ == '__main__':
    n, c = input().split()
    m = 1
    s = 1
    while int(n) >= s:
        m += 2
        s += 2 * m
    s -= 2 * m
    m -= 2
    for i in range(m, 0, -2):
        r = ""
        for j in range(0, i):
            r += c
        for j in range(0, (m - i) // 2):
            r = " " + r
        print(r)

    for i in range(3, m + 1, 2):
        r = ""
        for j in range(0, i):
            r += c
        for j in range(0, (m - i) // 2):
            r = " " + r
        print(r)
    print(int(n) - s)