PTA L1-002 打印沙漏 （python版本）

L1-002 打印沙漏 （20 分）

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

参考代码：

#coding:utf-8
str=raw_input();
c=len(str)-1;
num=0;
for i in str:
    if i==' ':
        break;
    num=num*10+int(i);
p=1;
num-=1;
k=3;
while (num-2*k)>=0:
    p+=2;
    num-=2*k;
    k=k+2;
ni=p;
while ni>0:
    if ni<p:
        print ((p-ni)/2-1)*' ',;
    print ni*str[c];
    ni-=2;
ni=3;
while ni<=p:
    if ni<p:
        print ((p-ni)/2-1)*' ',;
    print ni*str[c];
    ni+=2;
print num;