Leetcode1415-长度为 n 的开心字符串中字典序第 k 小的字符串(Python3实现)

class Solution:
    def getHappyString(self, n: int, k: int) -> str:
        def add(array):
            #因为k值大小必定是小于等于所有情况数的
            #所以，当array长度为1时，此时array值必定只能为‘a‘或‘b‘，只要进行直接更新即可
            if len(array)==1:
                if array[-1]==‘a‘:
                    return [‘b‘]
                else:
                    return [‘c‘]
            #当array长度大于1时，就分情况进行讨论
            ##第1,2种情况：[a,b],[b,a],直接更新最后一位为‘c‘即可
            if (array[-2]==‘a‘ and array[-1]==‘b‘) or (array[-2]==‘b‘ and array[-1]==‘a‘):
                array[-1]=‘c‘
                return array
            #第3种情况：[c,a],直接更新最后一位为‘b‘即可
            elif array[-2]==‘c‘ and array[-1]==‘a‘:
                array[-1]=‘b‘
                return array
            #第4,5,6种情况:[a,c],[b,c],[c,b],其中[a,c]情况得到的最后一位必定是‘a‘,后两种的最后一位更新情况需要根据倒数第二位决定
            else:
                tmp=add(array[:-1])
                return tmp + [‘b‘] if  tmp[-1]==‘a‘ else tmp + [‘a‘]
        
        #字符串每增加一位长度，情况*2
        allCases=3*(2**(n-1))
        #判定n,k是否符合条件
        if allCases<k or n<1 or k<0:
           return ""
        #生成第一种情况的字符串
        arr=[‘a‘]
        for i in range(1,n):
            arr.append(‘a‘) if arr[-1]!=‘a‘ else arr.append(‘b‘)
        #循环k-1次得到第k次的结果
        for i in range(1,k):
            arr=add(arr)
        return ‘‘.join(arr)