PTA数据结构起步能力自测题-1：打印沙漏

题目要求

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

代码

#include <bits/stdc++.h>
using namespace std;
int main()
{
    int i,n,temp=0,cnt=1,col=1,space=0;
    char ch;
    bool flag=false;
    cin>>n>>ch;
    if(n==1){
        cout<<ch<<endl<<"0"<<endl;
        exit(0);
    }
    while(cnt<n){
        col+=2;
        cnt+=col*2;
    }
    cnt-=col*2;
    col-=2;
    temp=col;
    for(;abs(temp)<=col;temp-=2){
        if(flag&&abs(temp)==1) continue;
        if(!flag&&temp==1)  flag=true;
        space=col-abs(temp);
        for(i=0;i<space/2;i++){
            cout<<" ";
        }
        for(i=0;i<abs(temp);i++){
            cout<<ch;
        }
        cout<<endl;
    }
    cout<<n-cnt<<endl;
    return 0;
}

提交结果