具有给定数值的最小字符串
难度:中等
小写字符 的 数值 是它在字母表中的位置(从 1
开始),因此 a
的数值为 1
,b
的数值为 2
,c
的数值为 3
,以此类推。
字符串由若干小写字符组成,字符串的数值 为各字符的数值之和。例如,字符串 "abe"
的数值等于 1 + 2 + 5 = 8
。
给你两个整数 n
和 k
。返回 长度 等于 n
且 数值 等于 k
的 字典序最小 的字符串。
注意,如果字符串 x
在字典排序中位于 y
之前,就认为 x
字典序比 y
小,有以下两种情况:
x
是y
的一个前缀;- 如果
i
是x[i] != y[i]
的第一个位置,且x[i]
在字母表中的位置比y[i]
靠前。
示例 1:
输入:n = 3, k = 27
输出:"aay"
解释:字符串的数值为 1 + 1 + 25 = 27,它是数值满足要求且长度等于 3 字典序最小的字符串。
示例 2:
输入:n = 5, k = 73
输出:"aaszz"
贪心
思路:
根据题意判断可以得到结论,只要满足前面留存1,后边的数值越大越好,即可。
因此,我们从转换字符串结尾开始计算,每次都取能够取到最大的字符为结果,最后将所有的结果合并返回则可。
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n),其中 n n n 表示给定的数字 n n n。我们只需依次构造 n n n 个字符即可。
- 空间复杂度: O ( n ) O(n) O(n)。返回值长度数组为n。
class Solution:
def getSmallestString(self, n: int, k: int) -> str:
res = [0] * n
while n != 0:
now = min(k - n + 1, 26)
k -= now
n -= 1
res[n] = chr(96 + now)
return "".join(res)
贪心优化
思路:
根据上述思路进行改进,我们可以模拟先给每个
n
n
n 都分配一个最小字符
‘
a
’
‘a’
‘a’,则剩余数字为 k-n
,此时我们只需要计算能够填充多少个字符到
‘
z
′
‘z'
‘z′,即为 k-n // 25
,
25
25
25 为
z
−
a
z-a
z−a,此时我们需要判断 k-n // 25
的数量是否等于n,如果等于则说明全部都是‘z’,返回n * 'z'
即可,否则返回(n-z-1) * 'a' + chr(96 + 1 + (k - n) % 25) + z * 'z'
,其中 z
为 k-n // 25
。
复杂度分析:
- 时间复杂度: O ( 1 ) O(1) O(1),只进行了固定数量的基本计算。
- 空间复杂度: O ( 1 ) O(1) O(1)。常数个变量。
class Solution:
def getSmallestString(self, n: int, k: int) -> str:
z = (k - n) // 25
return z * 'z' if z == n else (n-z-1) * 'a' + chr(96 + 1 + (k - n) % 25) + z * 'z'
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/smallest-string-with-a-given-numeric-value