题目:2516. 每种字符至少取 K 个
思路:两端取出的元素“在满足要求的前提下”数量最少,等价于“剩下的中间部分”字符串最长。使用双指针求中间部分的最大长度,时间复杂度为0(n)。
C++版本:
class Solution {
public:
int takeCharacters(string s, int k) {
//记录字符串中每个字符出现的次数
vector<int> v(3,0);
for(auto x:s){
v[x-'a']++;
}
//不符合要求,提前结束
if(v[0]<k||v[1]<k||v[2]<k) return -1;
//左指针
int left=0;
//最大长度
int mx=0;
//i是右指针
for(int i=0;i<s.size();i++){
int x=s[i]-'a';
v[x]--;
//剩余元素数量不符合要求,左指针需要移动直到满足要求
while(v[x]<k){
v[s[left++]-'a']++;
}
//更新长度
mx=max(mx,i-left+1);
}
return s.size()-mx;
}
};
JAVA版本:
class Solution {
public int takeCharacters(String s, int k) {
//记录字符串中每个字符出现的次数
int[] v=new int[3];
for(int i=0;i<s.length();i++){
v[s.charAt(i)-'a']++;
}
//不符合要求,提前结束
if(v[0]<k||v[1]<k||v[2]<k) return -1;
//最大长度
int mx=0;
//左指针
int left=0;
//i是右指针
for(int i=0;i<s.length();i++){
int x=s.charAt(i) - 'a';
v[x]--;
//剩余元素数量不符合要求,左指针需要移动直到满足要求
while(v[x]<k){
v[s.charAt(left)-'a']++;
left++;
}
//更新长度
mx=Math.max(mx,i-left+1);
}
return s.length()-mx;
}
}