PAT 乙级 1027 打印沙漏 (20分)

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

#include <iostream>
using namespace std;
int main()
{
	int n,sum=0,m,count=0,i;
	char c;
	cin>>n>>c;
	for(int k=1;k<=n;k+=2)
	{
		sum+=k;
		if(2*sum-1==n)
		{
			m=k;
			break;
		}
			
		if(2*sum-1>n)
		{
			sum-=k;
			k-=2;
			m=k;
			break;
		}
		count++;
	}
	int f=count;
	for(i=m;i>=1;i-=2)
	{
		for(int j=count;j<f;j++) cout<<" ";
		for(int k=i;k>0;k--) cout<<c;
		cout<<endl;
		count--;
	}
	count=2;
	for(i=3;i<=m;i+=2)
	{
		for(int j=count;j<f;j++) cout<<" ";
		for(int k=i;k>0;k--) cout<<c;
		cout<<endl;
		count++;
	}
	cout<<n-2*sum+1<<endl;
	return 0;
}