给定一个全是大写字母字符串,可以最多选择k个字母把它替换为任意的字符,问可以得到最长的全部相同的字符串长度?
比如ABAB, k = 2, 我们可以把两个B替换为A, 得到一个长度为4的全部为A的字符串 , result =4
比如AABABBA, 我们可以中间的A替换为B, 得到AABBBBA, 得到长度为4的全是B的字符串,result=4
sol:
我们可以用滑动窗口来解决这个问题,需要保存的是窗口内最多字符的数量,如果剩下的字符数量个数 <=k,全都可以替换掉,否则,移动窗口的左边界来满足这个条件。
"窗口内最多字符的数" 这个我使用了遍历的方法来计算,这个应该是可以优化到O(1)的,没有继续想了。
int maxLetterCnt = 0;
public int characterReplacement(String s, int k) {
int left = 0;
int right = 0;
int len = s.length();
int res = Math.min(len, k);
int[] cnt = new int[26];
for (right = 0; right < s.length(); right++) {
char c = s.charAt(right);
cnt[c - 'A']++;
int size = right - left + 1;
if (size <= k) continue;
int maxCnt = getMaxLetter(cnt);
int remain = size - maxCnt;
if (remain <= k) {
res = Math.max(res, size);
} else {
while(true) {
char ch = s.charAt(left++);
cnt[ch - 'A']--;
maxCnt = getMaxLetter(cnt);
size = right - left + 1;
remain = size - maxCnt;
if (remain <= k) {
res = Math.max(res, size);
break;
}
}
}
}
return res;
}
int getMaxLetter(int[] cnt) {
int res = Integer.MIN_VALUE;
for (int i = 0; i < 26; i++) {
res = Math.max(res, cnt[i]);
}
return res;
}