PAT (Basic)1027 打印沙漏 （Python實現）

打印沙漏

本題要求你寫個程序把給定的符號打印成沙漏的形狀。例如給定17個“*”，要求按下列格式打印

*****
 ***
  *
 ***
*****

所謂“沙漏形狀”，是指每行輸出奇數個符號；各行符號中心對齊；相鄰兩行符號數差2；符號數先從大到小順序遞減到1，再從小到大順序遞增；首尾符號數相等。

給定任意N個符號，不一定能正好組成一個沙漏。要求打印出的沙漏能用掉盡可能多的符號。

輸入格式：

輸入在一行給出1個正整數N（≤1000）和一個符號，中間以空格分隔。

輸出格式：

首先打印出由給定符號組成的最大的沙漏形狀，最后在一行中輸出剩下沒用掉的符號數。

輸入樣例

19 *

輸出樣例：

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

---

分析：

觀察每個沙漏，看每個沙漏由多少個符號組成。發現它們的遞增規律是1、7、17、...、2*k^2-1,那么找出這個K就能把沙漏打印出來了。

---

代碼

import math

N, sign = input().split(" ")
N = int(N)
k = int(math.sqrt((N+1)/2))
# 求出剩下的符號數
surplus = N-2*k**2+1

# 打印沙漏
for i in range(k):
    print(" "*i, end="")
    print(sign*(2*(k-i)-1))
for i in range(2,k+1):
    print(" "*(k-i), end="")
    print(sign*(2*i-1))
print(surplus)