给你一个由字符 '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 <= 105
s
仅由字母'a'
、'b'
、'c'
组成0 <= k <= s.length
思路分析:
如果我们正着求,想用滑动窗口的思路的话则需要4个指针。因此,我们可以反正来求,最多可以减少多少字母来满足条件。以上图为例,设置两个窗口指针,l,r。r为右指针,l为左指针,当r指针右滑不满足条件时,则一直滑动左指针直到满足条件为止。下面的代码在第二个for循环是已经把不满足答案的给舍去了,所以下面的while循环一定能够找到满足num[ord(s[r])-97]<0:的情况,因此不用添加额外的判断
代码如下
class Solution(object):
def takeCharacters(self, s, k):
num=[0]*3
for i in s:
num[ord(i)-97]+=1
for i in range(len(num)):
num[i]-=k
if num[i]<0:
return -1
l=0
r=0
ans=0
for r in range(len(s)):
num[ord(s[r])-97]-=1
while num[ord(s[r])-97]<0:
num[ord(s[l])-97]+=1
l+=1
ans=max(r-l+1,ans)
return len(s)-ans