长度为 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
测试结果
执行用时:28 ms, 在所有 Python3 提交中击败了 95.35% 的用户
内存消耗:14.7 MB, 在所有 Python3 提交中击败了 100.00% 的用户
说明
算法题来源:力扣(LeetCode)