PTA打印沙漏

#include <stdio.h>
#include <math.h>

int main(int argc, const char * argv[]) {
    int input;
    char c;
    int need=1;
    int count=3;
    scanf("%d %c",&input,&c);
    while (need<=input) {
        need=need+2*count;
        count+=2;
    }
    need=need-2*(count-2);
    int line=count-4;
    for (int i=1; i<=line;i+=2) {
        for (int k=2; k<=i; k+=2) {
            printf(" ");
        }
        for (int j=line;j>=i;j--) {
            printf("%c",c);
            if (j==i) {
                printf("\n");
            }
        }
    }
    int maxSpace=(line+1)/2-1;
    for (int i=line-3; i>=0; i-=2) {
        for (int k=1; k<maxSpace; k++) {
            printf(" ");
        }
        for (int j=i; j<line; j++) {
            printf("%c",c);
        }
        maxSpace--;
        printf("\n");
    }
    
    printf("%d\n",input-need);
    return 0;
}