PAT乙级 1006 换个格式输出整数 (15分)

让我们用字母 B 来表示“百”、字母 S 表示“十”，用 12...n 来表示不为零的个位数字 n（<10），换个格式来输出任一个不超过 3 位的正整数。例如 234 应该被输出为 BBSSS1234，因为它有 2 个“百”、3 个“十”、以及个位的 4。

输入格式：

每个测试输入包含 1 个测试用例，给出正整数 n（<1000）。

输出格式：

每个测试用例的输出占一行，用规定的格式输出 n。

输入样例1：

234

输出样例1：

BBSSS1234

输入样例2：

23

输出样例2：

SS123

参考代码：

#include <iostream>
using namespace std;
int main(){
    int n;
    cin>>n;
    int num[5] = {0}, cnt = 0;
    while(n){
        num[cnt++] = n % 10;
        n /= 10;
    }
    if(cnt == 1){
        for(int i = 1; i <= num[0]; i++)
            cout<<i;
        cout<<endl;
    }
    else if(cnt == 2){
        for(int i = 1; i <= num[1]; i++)
            cout<<'S';
        for(int i = 1; i <= num[0]; i++)
            cout<<i;
        cout<<endl;
    }
    else if(cnt == 3){
        for(int i = 1; i <= num[2]; i++)
            cout<<'B';
        for(int i = 1; i <= num[1]; i++)
            cout<<'S';
        for(int i = 1; i <= num[0]; i++)
            cout<<i;
        cout<<endl;
    }
    return 0;
}