1027. 打印沙漏(20) 两个测试数据没有通过

pat乙级的题，总共四组测试数据
有两组测试样点过不了，不知道为什么，改了好几次没找到原因，求解答

 #include <bits/stdc++.h>
using namespace std;
int num[1055];
int g[1055];
void rowNum()               ///该函数作用：获得当前沙子数能打印出的最大沙漏层数
{
    num[1] = 1;         ///初始沙漏值
    g[1] = 0;                       ///该数组记录用剩下的沙子 
    int cnt = 7;        ///由于是从第三层开始算，第一层沙子数是1，要打印第三层沙子最小的数是7,
    int ans = 5;             ///第三层在加一层，最大层的沙子数是原来沙子数+2
    for(int i=2; i<1055; i++)
    {
        if( i == cnt)
        {
            g[i] = 0;
            cnt+=ans; 
            cnt+=ans; 
            ans+=2;
            num[i] = num[i-1]+2;
        }
        else
        {
            g[i] = g[i-1] + 1;
            num[i] = num[i-1];
        }
    }
}
int main()
{
    rowNum();
    int n;
    char c; 
    cin>>n>>c;
    int k = num[n];               ///获取总层数
    if(k ==1)
        cout<<c<<endl;
    else
    {
        int h=k;
        for(int i=(k+1)/2; i>=1; i--)            ///正三角有多少行
        {
            ///输出空格数
            int res =(k-h)/2;

            for(int j=0; j<res; j++)
                cout<<" ";
            for(int j=h; j>=1; j--)
                cout<<c;
            h=h-2;
           /// if(h>0)
                cout<<endl;
        }
        ///输出倒立的三角形
        h=1;

        for(int i=(k+1)/2-1; i>=1; i--)         ///倒三角有多少行
        {
            int res = (k-h)/2;
            for(int j=1; j<res; j++)
                cout<<" ";
            h+=2;
            for(int j=1; j<=h; j++)
                cout<<c;
            if(h<k)
                cout<<endl;
        }
        if(g[n] !=0 )
        {
            cout<<endl;
            cout<<g[n];
        }
    }
    return 0;
}

https://blog.csdn.net/ice_camel/article/details/42226067