团体程序设计天梯赛 L1-002 打印沙漏

打印沙漏

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

题解:
首先打表记录层数符号个数的前缀和，从层数1开始，相同个数的符号所在的层数相同。例如，1个符号为1层，3个符号的为第二层。
ceng[1]=1
ceng[2]=1+3x2
ceng[3]=1+3x2+5x2

之后根据所给的n求出层数，这里用到lowerbound函数
n可能正好满足所给的层数的符号个数，也有可能大
所以用一个flag来判断。
之后就是打印每层的符号

坑点：
要输出多出的符号个数
个数正好 要输出0

#include <bits/stdc++.h>

using namespace std;
int ceng[500];
int flag=false;
int n;
char c;
int main(void)
{
    ceng[1]=1;
    int num=3;
    for(int i=2;i<25;i++) 
    {
        ceng[i]=ceng[i-1]+num*2;
        num+=2;
    }
    cin>>n;
    getchar();
    cin>>c;

    int pos=lower_bound(ceng+1,ceng+25,n)-ceng;
    if(ceng[pos]!=n)
    {
        pos--;
        flag=true;
    }
    for(int i=pos;i>=1;i--)
    {
        for(int k=pos-i;k>=1;k--)
            cout<<" ";
        for(int j=2*i-1;j>=1;j--)
            cout<<c;
        cout<<endl;
    }
    for(int i=2;i<=pos;i++)
    {
        for(int k=pos-i;k>=1;k--)
            cout<<" ";
        for(int j=2*i-1;j>=1;j--)
            cout<<c;
        cout<<endl;
    }
    if(flag)
        cout<<n-ceng[pos]<<endl;
    else
        cout<<0<<endl;
    return 0;
}