leetcode(98)_1415_medium_长度为 n 的开心字符串中字典序第 k 小的字符串_python

本文链接：https://blog.csdn.net/qq_40137656/article/details/121151818

长度为 n 的开心字符串中字典序第 k 小的字符串

题目描述：
一个「开心字符串」定义为：
仅包含小写字母 [‘a’, ‘b’, ‘c’].
对所有在 1 到 s.length - 1 之间的 i ，满足 s[i] != s[i + 1] （字符串的下标从 1 开始）。
比方说，字符串 “abc”，“ac”，“b” 和 “abcbabcbcb” 都是开心字符串，但是 “aa”，“baa” 和 “ababbc” 都不是开心字符串。
给你两个整数 n 和 k ，你需要将长度为 n 的所有开心字符串按字典序排序。
请你返回排序后的第 k 个开心字符串，如果长度为 n 的开心字符串少于 k 个，那么请你返回空字符串。
示例：
输入：n = 1, k = 3
输出：“c”
解释：列表 [“a”, “b”, “c”] 包含了所有长度为 1 的开心字符串。按照字典序排序后第三个字符串为 “c” 。
提示：

1 <= n <= 10
1 <= k <= 100

解法

我们先分析共有多少种排列方式，第一个字符显然有三种选择；后面的字符因为不能和前一个相同，所以有两种选择，所以对于长度为 n 的开心字符串，共有 3*2^(n-1) 种排列方式。因此我们可能逐个确定字符。举例来讲，n=3，k=5，由于第二个和第三个字符共有四种情况，而 k = 5，这样一来，第一个字符只能是 b；确定 b 后，可以继续确定第二个字符，此时只需要找到后面字符组成的第 1 小的字符串，因此第二个字符是 a，第三个字符也是要找到第 1 小，同时不能和 a 相同，因此是 b。

代码

class Solution:
    def getHappyString(self, n: int, k: int) -> str:
        num = pow(2, n - 1)
        if k > 3 * num:
            return ""
        k -= 1  # 将 1 ~ m 降为 0 ~ m-1，与索引从 0 开始相匹配。否则n=1，k=3，k//n=3会越界
        candidate_ch = ['a', 'b', 'c']
        ch_list = [candidate_ch[k // num]]  # 先确定首字符
        k %= num  
        for i in range(1, n):
            num //= 2
            ch_list.append(self.func(candidate_ch, ch_list[-1], k // num))
            k %= num
        return "".join(ch_list)
            

    def func(self, candidate_ch, pre_ch, k):  # 在前一个字符为 pre_ch 的情况下，找下一个第 k 小的字符。k = 0 或 1
        for ch in candidate_ch:
            if ch != pre_ch:
                if k == 0:
                    return ch
                k -= 1