打印沙漏(pta)

#include<stdio.h>
int main(){
    int N,H,h,a,i=1;
    char ch;
    scanf("%d %c",&N,&ch);
    N--;
    for(h=1;N>i*2;h+=2){
        i+=2;
        N-=i*2;
    }
    if(N<0){
        h-=2;
        N+=i*2;
        i-=2;
    }
    H=h;
    for(a=i;h>0;h--){
        for(i=(H-a)/2;i>0;i--){
            printf(" ");
        }
        for(i=0;i<a;i++){
            printf("%c",ch);
        }
        if(h<=H/2+1){
            a+=2;
        }else{
            a-=2;
        };
        printf("\n");
    }
    printf("%d",N);
    return 0;
}