PAT乙级-B1027 打印沙漏（20）

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

#include<iostream>
#include<cmath>
#include<string>
using namespace std;
int main(){
    int n, r;
    char c;
    cin >> n >> c;
    //可知1，3，5……的数列，前n项和sn = n^2，而沙漏的和 bn = sn*2 - 1
    //r表示沙漏的一半行数（向上取整）
    r = sqrt((n+1)/2);
    int t = r;
    string s;
    //开始输出，从r行开始输出至1之前，记得插入空格
    while(t > 1){
        s.insert(0,1+2*(t-1),c);
        s.insert(0,r - t, ' ');
        cout << s<<endl;
        s.clear();
        t--;
    }
    //从1开始至r输出
    while(t <= r){
        s.insert(0,1+2*(t-1), c);
        s.insert(0,r - t, ' ');
        cout << s << endl;
        s.clear();
        t++;
    }
    cout << n - 2*r*r + 1 << endl;
    return 0;
}