L1-002 打印沙漏(20分)

原题链接

第一处的空格：i * strlen(str)
第二处的空格：(sum-i-1)*strlen(str)

★,°:.☆(￣▽￣)/$:.°★ 。撒花

#include<bits/stdc++.h>
#include<cstring>
#include<vector>
using namespace std;
int main()
{
    int n;
    char str[2];
    cin>>n>>str;
    int k=(int)sqrt((n+1)/2)-1;//上面k行
    int sum=k*2+1;
    int res=n+1-2*(k+1)*(k+1);
    for(int i=0;i<sum;i++)
    {
        if(i<=sum/2)
        {
        for(int j=0;j<i;j++)
            cout<<" ";
        for(int j=0;j<sum-2*i;j++)
            cout<<str;
        cout<<endl;
        }
        else
        {
        for(int j=0;j<sum-i-1;j++)
            cout<<" ";
        for(int j=0;j<2*i-sum+2;j++)
            cout<<str;
        cout<<endl;
        }
    }
    cout<<res<<endl;
    return 0;
}