PTA练习 L1-002 打印沙漏

#include<iostream>
using namespace std;
int main()
{
	char N;
	int sum;
	int link;
	cin >> sum >> N;
	//计算一半的行数
	for (int i = 0; 2 * i * i + 4 * i + 1 <= sum; i++)
	{
		link = i;
	}
	//打印上半
	//对齐
	for (int i = 0; i < link; i++)
	{
		for (int n = 0; n < i; n++)
		{
			cout << " ";
		}
		for (int m = 2 * link + 1-i*2; m >= 1; m --)
		{
			cout << N;
		}
		cout << endl;
	}
	//打印中间部分
	for (int i = 0; i < link; i++)
	{
		cout << " ";
	}
	cout << N << endl;
	//打印下半
	//对齐
	for (int i = link; i > 0; i--)
	{

		for (int n = i-1; n >= 1; n--)
		{
			cout << " ";
		}
		for (int m = 2*i-1; m <= 2*link+ 1; m++)
		{
			cout << N;
		}
		cout << endl;
	}

	cout << sum - (2 * link * link + 4 * link + 1);
}