PAT 乙级 1027. 打印沙漏(20) C++版

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

*****
***
​ *
***
*****

所谓“沙漏形状”,是指每行输出奇数个符号;各行符号中心对齐;相邻两行符号数差2;符号数先从大到小顺序递减到1,再从小到大顺序递增;首尾符号数相等。 给定任意N个符号,不一定能正好组成一个沙漏。要求打印出的沙漏能用掉尽可能多的符号。

输入格式:

输入在一行给出1个正整数N(<=1000)和一个符号,中间以空格分隔。

输出格式:

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

输入样例:

19 *

输出样例:

*****
***
​ *
***
*****
2

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#include <iostream>
#include <algorithm>
#include <cmath>

using namespace std;

int main() {
int n = 0;
char c;
cin >> n;
cin >> c;

int w = sqrt((n + 1) / 2);
for (int i = 0; i < 2 * w - 1; i++) {
for (int j = 0; j < 2 * w - 1; j++) {
if ((i > j && i + j < 2 * w - 2) || (i < j && i + j > 2 * w - 2)) {
if (i > j && i + j < 2 * w - 2)
cout << " ";
} else {
cout << c;
}
}
cout << endl;
}

cout << n - 2 * w * w + 1 << endl;
return 0;
}