PAT1027 打印沙漏

#include <stdio.h>
#include <string.h>

int shalou(size_t size, char s)
{
    int top = 0;
    if(size == 0)
    	return 0;
    for(int i = 3; i <= 10000; i += 2)
    {
        top += i;
        if(top*2+1 > size)
        {
            top -= i;
            size = top == 1 ? 1 : top * 2 + 1;
            break;
        }
    }
    char arr[1000][1000];
    int  cnt = 0, index = 0;
	for(int i = 1; cnt < top + 1; i+=2)
	{
		cnt+=i;
		for(int j = 0; j < i; j++)
			arr[index][j] = s;
		index++;
	}
	for(int i = index - 1; i >= 0; i --)
	{
		for(int j = 0; j < (strlen(arr[index - 1])-strlen(arr[i]))/2; j++)
			printf(" ");
		printf("%s\n", arr[i]);
	}
	for(int i = 1; i < index; i++)
	{
		for(int j = 0; j < (strlen(arr[index - 1])-strlen(arr[i]))/2; j++)
			printf(" ");
		printf("%s\n", arr[i]);
	}
	return size;
}

int main()
{
	int n;
	char c;
	scanf("%d %c", &n, &c);
	int la = n - shalou(n, c);
	printf("%d\n", la); 
    return 0;
}