PAT-1027 打印沙漏

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <iomanip>
using namespace std;

int main()
{
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt", "w", stdout);
    int n = 0, m = 0;
    char c;
    cin >> n >> c;
    m = (n-1)/2;

    int sum = 0, i;
    for (i = 3; ; i += 2)
    {
        sum += i;
        if (sum > m)
            break;
    }
    sum -= i;
    i -= 2;

    for (int j = i; j >= 3; j -= 2)
    {
        for (int k = 0; k < (i-j)/2; k++)
            cout << " ";
        for (int k = j-1; k >=0; k--)
            cout << c;
        cout << endl;
    }

    for (int k = 0; k < (i-1)/2; k++)
        cout << " ";
    cout << c << endl;

    for (int j = 3; j <= i; j += 2)
    {
        for (int k = 0; k < (i-j)/2; k++)
            cout << " ";
        for (int k = 0; k < j; k++)
            cout << c;
        cout << endl;
    }

    cout << n - 1 - 2*sum;
    return 0;
}