PAT B1027 打印沙漏 (20point(s))

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

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

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

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

输入格式:

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

输出格式:

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

输入样例:

19 *

输出样例:

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

• 思路：
  可以组成沙漏的字符数组为：1、1 + 3 * 2、1+（3 + 5）* 2 ... 即：1 * 2 - 1、(1 + 3) * 2 - 1、（1 + 3 + 5）* 2 - 1 ...
  递推公式为：Di = 2 * i² - 1；
  step1：预处理出标记数组mark[]和line[]，mark[n]为Di（可构成沙漏的字符数，类似于"向下取整")；line[n]为i（第i项）；2 * line[n] - 1 为行数，也为第一行字符数
  step2：格式化输出：第1到line[n]行，空格数每行+1，字符数每行-2；第line[n]+1行到第2*line[n]-1行，空格数每行-1，字符数每行+2

• code：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2010;
int mark[maxn], line[maxn];
void init(){
	for(int i = 1; i < 25; ++i){
		int now = 2 * i * i - 1, next = 2 * (i + 1) * (i + 1) - 1;
		for(int j = now; j < next; ++j){
			line[j] = i;
			mark[j] = now;
		}
	}
}
int main(){
	int n;
	char c;
	init();
	scanf("%d %c", &n, &c);
	int sum_line = 2 * line[n] - 1;
	int space_cnt = 0, char_cnt = sum_line;
	for(int i = 0; i < sum_line; ++i){
		for(int j = 0; j < space_cnt; ++j) printf(" ");
		for(int j = 0; j < char_cnt; ++j) printf("%c", c);
		printf("\n");
		if(i < line[n]-1){ //1~line[n]行 
			space_cnt++; char_cnt -= 2;
		}else{ //line[n]+1到2*line[n]-1 
			space_cnt--; char_cnt += 2;			
		}
	}
	printf("%d", n - mark[n]);
	return 0;
}