今天抽时间又刷一道leetcode
题目:2904. 最短且字典序最小的美丽子字符串最短且字典序最小的美丽子字符串2904. 最短且字典序最小的美丽子字符串
给你一个二进制字符串
s
和一个正整数k
。如果
s
的某个子字符串中1
的个数恰好等于k
,则称这个子字符串是一个 美丽子字符串 。令
len
等于 最短 美丽子字符串的长度。返回长度等于
len
且字典序 最小 的美丽子字符串。如果s
中不含美丽子字符串,则返回一个 空 字符串。对于相同长度的两个字符串
a
和b
,如果在a
和b
出现不同的第一个位置上,a
中该位置上的字符严格大于b
中的对应字符,则认为字符串a
字典序 大于 字符串b
。
- 例如,
"abcd"
的字典序大于"abcc"
,因为两个字符串出现不同的第一个位置对应第四个字符,而d
大于c
。
示例 1:
输入:s = "100011001", k = 3 输出:"11001" 解释:示例中共有 7 个美丽子字符串: 1. 子字符串 "100011001" 。 2. 子字符串 "100011001" 。 3. 子字符串 "100011001" 。 4. 子字符串 "100011001" 。 5. 子字符串 "100011001" 。 6. 子字符串 "100011001" 。 7. 子字符串 "100011001" 。 最短美丽子字符串的长度是 5 。 长度为 5 且字典序最小的美丽子字符串是子字符串 "11001" 。示例 2:
输入:s = "1011", k = 2 输出:"11" 解释:示例中共有 3 个美丽子字符串: 1. 子字符串 "1011" 。 2. 子字符串 "1011" 。 3. 子字符串 "1011" 。 最短美丽子字符串的长度是 2 。 长度为 2 且字典序最小的美丽子字符串是子字符串 "11" 。示例 3:
输入:s = "000", k = 1 输出:"" 解释:示例中不存在美丽子字符串。
提示:
1 <= s.length <= 100
1 <= k <= s.length
题解:滑动窗口 + 双指针
class Solution:
def shortestBeautifulSubstring(self, s: str, k: int) -> str:
if s.count('1') < k:
return ''
ans = s
temp_cnt , left = 0, 0
for right, c in enumerate(s):
temp_cnt += int(c)
while temp_cnt > k or s[left] == '0':
temp_cnt -= int(s[left])
left += 1
if temp_cnt == k:
t = s[left: right + 1]
if len(t) < len(ans) or len(t) == len(ans) and t < ans:
ans = t
return ans
时间复杂度:O(n2)
空间复杂度:O(1)