PAT 乙级1027打印沙漏

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

代码： 

#include <iostream>
using namespace std;

int main() {
	int n;
	char s;
	int max = 1;
	int temp = 0;
	int sum = 0;
	int cnt = 0;
	while (~scanf("%d %c", &n, &s)) {
		for (int i = 3; i < 1000; i += 2) {
			temp += i;
			sum = temp * 2 + 1;
			if (sum > n)
				break;
			else
				max = i;
		}
		for (int i = max; i > 0; i -= 2) {
			for (int a = (max - i) / 2; a > 0; a--) {
				printf(" ");
			}
			for (int b = i; b > 0; b--) {
				printf("%c", s);
				cnt++;
			}
			cout << endl;
		}
		for (int i = 3; i <= max; i += 2) {
			for (int a = (max - i) / 2; a > 0; a--) {
				printf(" ");
			}
			for (int b = i; b > 0; b--) {
				printf("%c", s);
				cnt++;
			}
			cout << endl;
		}
		cout << n - cnt;
	}

	return 0;
}