PAT 1027 打印沙漏

传送门
找规律╮(╯-╰)╭
本题要求你写个程序把给定的符号打印成沙漏的形状。例如给定17个“*”，要求按下列格式打印

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

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

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

输入格式：

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

输出格式：

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

输入样例：

19 *
输出样例：

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

2

#include <iostream>
#include <string.h>
#include <stdio.h>

using namespace std;

int main()
{
    int n,N;
    char ch;
    scanf("%d %c",&N,&ch);
    for(int i=0;; i++)
    {
        if((N-(2*(i+1)*(i+1)-1))==0)
        {
            n = i+1;
            break;
        }
        else if((N-(2*i*i-1))*(N-(2*(i+1)*(i+1)-1))<0)
        {
            n = i;
            break;

        }
    }
    for(int i=1; i<=n; i++)
    {
        for(int j=1; j<i; j++)
            printf(" ");
        for(int j=1; j<=2*(n-i+1)-1; j++)
        {

            if(j==2*(n-i+1)-1)
                printf("%c\n",ch);
            else
                printf("%c",ch);
        }
    }
    for(int i=1; i<n; i++)
    {
        for(int j=i+1; j<n; j++)
            printf(" ");
        for(int j=1; j<=2*i+1; j++)
        {

            if(j==2*i+1)
                printf("%c\n",ch);
            else
                printf("%c",ch);
        }
    }
    printf("%d",N-(2*n*n-1));
    return 0;
}