网易笔试题---小易的字典

 一题目

解答：

排列组合，n个'a'和m个'z'，能组成(m+n)!/(m!)/(n!)个单词

思路：

1. 假设第一个字符为a，则剩下n-1个'a'和m个'z'组成的子序列只能构成(m+n-1)!/(m!)/(n-1!)个单词，且是字典中前(m+n-1)!/(m!)/(n-1!)个单词。
2. 比较k和(m+n-1)!/(m!)/(n-1!)，若k小，说明k是前(m+n-1)!/(m!)/(n-1!)个单词，则第一个字符必为'a'。子问题化为在子序列(n-1个'a'和m个'z')找到第k个单词
3. 若k大，则说明第一个字符必为'z',单词是以'z'开头的单词中的第k-(m+n-1)!/(m!)/(n-1!)个。子问题化为在子序列(n个'a'和m-1个'z')找到第k-count(n+m-1,m-1)个单词。

eg:n=2,m=2,k=5

1. 假设第一个字符为a,则剩下1个a,2个z只能构成3个单词，且是字典中前3个单词(aamm,amam,amma)
2. k>3，则第一个字符必为z。原问题化为在n=2,m=1,k=2，即在剩下2个a，1个z中找到第2个单词

   def paixu(n,m):
       res=1
       for i in range(n+1,n+m+1):
           res*=i
       for j in range(1,m+1):
           res//=j
       return res
   
   
   def findk(n,m,k):
       result=''
       if k>paixu(n,m):
           return -1
       else:
           while m and n:
               res=paixu(n-1,m)
               if k>res:
                   k-=res
                   result+='z'
                   m-=1
               else:
                   result+='a'
                   n-=1
           result+='a'*n
           result+='z'*m
       return result
   
   n,m,k=list(map(int,input().split()))
   p=findk(n,m,k)
   print(p)