CF1032B Personalized Cup

#include<bits/stdc++.h>
using namespace std;
int a,b;
string s;
void print(int a,int b,int p)//打印结果 
{
    for(int i=1;i<=a;i++)
    {
        for(int j=1;j<=b;j++) printf("%c",s[p++]);
        printf("\n");
    }
}
signed main()
{
    cin>>s;//输入字符串
    //判断 
    if(s.size()%20==0)
    {
        a=s.size()/20, b=20;
        printf("%d %d\n",a,b);
        print(a,b,0);
    }
    else
    {
        a=s.size()/20+1;
        if(s.size()%a==0)
        {
            b=s.size()/a;
            printf("%d %d\n",a,b);
            print(a,b,0);
        }
        else
        {
            b=s.size()/a+1;
            printf("%d %d\n",a,b);
            int cnt=0, flag=0;
            for(int i=1;i<=a;i++)
            {
                if(flag==0 && (s.size()-cnt)%(b-1)==0) b--, flag=1;
                for(int j=1;j<=b;j++) printf("%c",s[cnt++]);
                if(flag) printf("*");
                printf("\n");
            }
        }
    }
    return 0;
}