【题解】ABC202D aab aba baa

题目传送门 AtCoder原题

---

$\text{Solution}$

具体实现

经典数学问题

样例

2 2 4

明显地，当只有 $2$ 个 $\text{a}$ 和 $2$ 个 $\text{b}$ 时，只有 $C_4^2=6$ 种可能。

怎么找呢？

由于第一位对字典序的影响最大，故先考虑第一位。

当第一位是 $\text{a}$ 时，有 $3$ 种可能，明显的，$\text{a}$ 开头的字典序应更靠前，而 $3<4$，故不可能第一位是 $\text{a}$，则第一位是 $\text{b}$，注意此时需要减去 $\text{a}$ 开头的情况数。

以此类推，我们就可以知道每一位的答案。

AC代码 洛谷记录 AT记录

$6ms/3.56MB$

#include<bits/stdc++.h>
using namespace std;
long long c[105][105];//预处理组合数，不开longlong见祖宗
int a,b;
long long k;
void init(){//DP预处理 证明见下方
    c[0][0]=1;
    for(int i=1;i<=60;i++){
        c[i][0]=1;
        for(int j=1;j<=i;j++) c[i][j]=c[i-1][j]+c[i-1][j-1];
    }
}
int main(){
    init();
    cin>>a>>b>>k;
    int l=a+b;//字符串总长
    for(int i=1;i<=l;i++){
        long long n=c[l-i][max(a-1,0)];//假设这一位是a，接下来的情况数
        if (a==0) n=0;//防止下标越界
        if (k>n) k-=n,putchar('b'),b--;//如果a的情况数不够用，则这一位为b
        else putchar('a'),a--;//否则是a
    }
    return 0;
}

---

附：组合数等价于杨辉三角的证明