拼多多2020校招部分编程题合集第三题

3.

多多的电子字典

多多鸡打算造一本自己的电子字典，里面的所有单词都只由a和b组成。
每个单词的组成里a的数量不能超过N个且b的数量不能超过M个。
多多鸡的幸运数字是K，它打算把所有满足条件的单词里的字典序第K小的单词找出来，作为字典的封面。

题目地址：

牛客网 - 找工作神器|笔试题库|面试经验|实习招聘内推，求职就业一站解决_牛客网 (nowcoder.com)

时间限制：C/C++ 1秒，其他语言2秒

空间限制：C/C++ 32M，其他语言64M

输入例子：

2 1 4

输出例子：

ab

例子说明：

满足条件的单词里，按照字典序从小到大排列的结果是
a
aa
aab
ab
aba
b
ba
baa

#include <iostream>
#include<string.h>
using namespace std;

int main() {
    unsigned long long n,m,k;
    cin>>n>>m>>k;
    unsigned long long arr[55][55];
    memset(arr,0,sizeof(arr));
    for(int i=0;i<=n;i++){
        for(int j=0;j<=m;j++){
            if(i==0&&j==0){
                arr[i][j]=0;
            }
            else if(i==0||j==0){
                arr[i][j]=1;
            }
            else{
                arr[i][j]=arr[i-1][j]+arr[i][j-1];
            }
        }
    }

    for(int i=0;i<=n;i++){
        for(int j=1;j<=m;j++){
            arr[i][j]+=arr[i][j-1];
        }
    }
    for(int j=0;j<=m;j++){
        for(int i=1;i<=n;i++){
            arr[i][j]+=arr[i-1][j];
        }
    }
    
    // for(int i=0;i<=n;i++){
    //     for(int j=0;j<=m;j++){
    //         cout<<arr[i][j]<<" ";
    //     }
    //     cout<<endl;
    // }
    int nn=n,mm=m;
    while(k>0){
        if(nn>0){
            if(k<=arr[nn-1][mm]+1){
                printf("a");
                nn--;
            }
            else{
                printf("b");
                k-=(arr[nn-1][mm]+1);
                mm--;
            }
            k--;
        }
        else if(nn==0&&mm>0){
            printf("b");
            k--;
            mm--;
        }
    }
    return 0;
}
// 64 位输出请用 printf("%lld")