PAT(Basic Level)_1027_打印沙漏

#include<stdio.h>

int main(){
    int N;
    char ch;
    scanf("%d %c",&N,&ch);

    int k=1;
    while(2*k*k-1<=N) k++;
    k--;

    for(int i=k;i>0;i--){
        for(int j=k-i;j>0;j--){
            putchar(' ');
        }
        for(int j=2*i-1;j>0;j--){
            putchar(ch);
        }
        putchar('\n');
    }
    for(int i=2;i<=k;i++){
        for(int j=k-i;j>0;j--){
            putchar(' ');
        }
        for(int j=2*i-1;j>0;j--){
            putchar(ch);
        }
        putchar('\n');
    }
    printf("%d",N-2*k*k+1);

    return 0;
}