PAT--陈越、何钦铭-数据结构-起步能力自测题--自测-1 打印沙漏 (20point(s))

题目链接：https://pintia.cn/problem-sets/17/problems/260

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int N;
	char c;
	cin>>N;
	cin>>c;
	int sum = 1,num = 0, count = 1; 
	while(sum <= N)
	{
		num++;
		count = count + 2;
		sum += count*2;
	}
	//cout<<num<<endl;  //测试
	for(int i = num-1; i >= 0; i--)
	{
		for(int k = 1;k < (num-i); k++)
		{
			cout<<" ";
		}
		for(int j = 1;j <= 2*i+1;j++)
		{
			cout<<c;
		}
		cout<<endl;
	}
	for(int i = 1; i < num; i++)
	{
		for(int k = (num-i-1); k>0; k--)
		{
			cout<<" ";
		}
		for(int j = 1;j <= 2*i +1; j++)
		{
			cout<<c;
		}
		cout<<endl;
	}
	cout<< N-(sum - count*2)<<endl;
	return 0;
}