PAT 1027. 打印沙漏

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

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

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

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

输入格式：

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

输出格式：

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

19 *

输出样例：

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

Python2.7代码实现：
测试用例：普通输入有剩余 19 *普通输入无剩余 17 *
最小输入 1 *
最大输入 1000 *
问题注意：1、普通输入无剩余注意0也要输出 2、输出字符后不需要继续输出空格，否则被判定格式错误。
__author__ = 'wilma'
import sys

inputString = raw_input()
inputList = inputString.split()
num = int(inputList[0])
sig = inputList[1]
nlist = [n for n in range(num) if (2*n*n-1 <= num)]
# print nlist

if len(nlist) == 1: #判断最小输入1
print sig
print num - 1
sys.exit()

nlist2 = [(2*n-1) for n in nlist[1:]]
numLast = nlist2[-1]

nlist2.reverse()

for x in nlist2:
if x == numLast:
print x*sig
else:
print ((numLast-x)/2-1)*' ', x*sig

nlist2.reverse()

for x in nlist2[1:]:
if x == numLast:
print x*sig
else:
print ((numLast-x)/2-1)*' ', x*sig #此处注意有坑，在打印了字符之后是不需要打印空格的

print num - 2*nlist[-1]*nlist[-1] + 1