PTA L1-002 打印沙漏 (20分)

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

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

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

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

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

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

输入样例:

19 *

输出样例:

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

#include<iostream>
#include<math.h>
using namespace std;

int main(){
    int n;    
    char c;
    cin>>n>>c;
    if (n == 1){  //测试点1
    	cout<<c<<endl;
    	cout<<0;
    	return 0;
	}
	
    int result=0;
    int i = 1;
    int j = 0;
    int surplus = 0;
    while(true){
        result = pow(i,2)-2*pow(j,2);
 //       cout<<result<<endl;
        if(n-result<=0){
        	i-=2;
         	j-=1;
            break;}
        i+=2;
        j+=1;
    }
    int temp_i = i;
    for(;temp_i>0;temp_i-=2){
    	int temp = temp_i;
    	if(i - temp>0){
		  int temp_cnt = (i - temp)/2;
		  for (;temp_cnt>0;temp_cnt--){
    		cout<<" ";
	    	}
		}
		
    	for (;temp>0;temp--){
    		cout<<c;
		}
		cout<<endl;
	}
     	
	for(int p = 3;p <= i;p+=2){
    	int temp = p;
    	if(i - temp>0){
		  int temp_cnt = (i - temp)/2;
		  for (;temp_cnt>0;temp_cnt--){
    		cout<<" ";
	    	}
		}
		
    	for (;temp>0;temp--){
    		cout<<c;
		}
		cout<<endl;
	}
	
    surplus = n - (pow(i,2)-2*pow(j,2));
 //   cout<<i<<" "<<j<<endl;
    cout<<surplus;
    return 0;
}