1027 打印沙漏

1027 打印沙漏

/**
 * 1.解题思路：每个沙漏都是从最中间一行行向上下分别扩展一行，
 *            每次扩展行都要比之前一层多2个符号，最中间一行只有 1 个符号，假设扩展的层数为 i
 *            则扩展出去的上边需要的所有符号个数为3 + 5 + 7 + … + (2i+1) = (3 + 2i + 1) * i / 2 = i * (i + 2)
 *            扩展出去的下边与上边同样多所以乘以2，加上最重要那一行1个符号，所以 总共需要2 * i * (i + 2) + 1个符号
 *            所以i从0到N，找满足(2 * i * (i + 2) + 1) > N的最小的 i，因为符号不能超过N，所以只能扩展出去 i-1 行
 *            用变量row表示从最中间一行需要扩展出去的行数，row = i – 1，接下来开始输出，上面的每一行
 *            对于扩展出去的第 i 层需要输出row – i个空格，接着输出i * 2 + 1个符号c和换行符；
 *            对于最中间一行，需要输出row – 1个空格、符号c和换行符；对于下面的每一行，对于扩展出去的第i层，需要输出row-i个空格
 *            接着输出i * 2 + 1个符号c和换行符，因为用掉的符号数为2 * row * (row + 2) + 1
 *            所以最后输出剩下没用掉的符号数为N – (2 * row * (row + 2) + 1)
 * 
 * 2.参考博客：https://www.liuchuo.net/archives/555 柳婼同学这个题的解题思路很清晰，推荐！
 */
#include <iostream>
using namespace std;
int main() {
    int N, row = 0;
    char c;
    cin >> N >> c;
    //找出需要扩展的行数row
    for (int i = 0; i < N; i++) {
        if ((2 * i * (i + 2) + 1) > N) {
            row = i - 1;
            break;
        }
    }
    //打印沙漏的上半部分
    for (int i = row; i >= 1; i--) {
        for (int k = row - i; k >= 1; k--) cout << " ";
        for (int j = i * 2 + 1; j >= 1; j--) cout << c;
        cout << endl;
    }
    for (int i = 0; i < row; i++) cout << " ";
    //打印沙漏中间的一行
    cout << c << endl;
    //打印沙漏的下半部分
    for (int i = 1; i <= row; i++) {
        for (int k = row - i; k >= 1; k--) cout << " ";
        for (int j = i * 2 + 1; j >= 1; j--) cout << c;
        cout << endl;
    }
    //输出剩余字符
    cout << (N - (2 * row * (row + 2) + 1));
    return 0;
}