title
给你一个由字符 'a'、'b'、'c' 组成的字符串 s 和一个非负整数 k 。每分钟,你可以选择取走 s 最左侧 还是 最右侧 的那个字符。
你必须取走每种字符 至少 k 个,返回需要的 最少 分钟数;如果无法取到,则返回 -1 。
示例 1:
输入:s = "aabaaaacaabc", k = 2
输出:8
解释:
从 s 的左侧取三个字符,现在共取到两个字符 'a' 、一个字符 'b' 。
从 s 的右侧取五个字符,现在共取到四个字符 'a' 、两个字符 'b' 和两个字符 'c' 。
共需要 3 + 5 = 8 分钟。
可以证明需要的最少分钟数是 8 。
示例 2:
输入:s = "a", k = 1
输出:-1
解释:无法取到一个字符 'b' 或者 'c',所以返回 -1 。
提示:
1 <= s.length <= 105s仅由字母'a'、'b'、'c'组成0 <= k <= s.length
solution
双指针。
class Solution:
def takeCharacters(self, s: str, k: int) -> int:
cnt = collections.Counter()
def check(cnt):
for c in "abc":
if cnt[c] < k:
return False
return True
n = len(s)
r = n
while not check(cnt):
r -= 1
if r < 0:
return -1
cnt[s[r]] += 1
ans = n - r
for l in range(n):
cnt[s[l]] += 1
while r < n and cnt[s[r]] > k:
cnt[s[r]] -= 1
r += 1
ans = min(ans, l + 1 + n - r)
return ans