一位老师正在出一场由 n 道判断题构成的考试,每道题的答案为 true (用 'T' 表示)或者 false (用 'F' 表示)。老师想增加学生对自己做出答案的不确定性,方法是 最大化 有 连续相同 结果的题数。(也就是连续出现 true 或者连续出现 false)。
给你一个字符串 answerKey ,其中 answerKey[i] 是第 i 个问题的正确结果。除此以外,还给你一个整数 k ,表示你能进行以下操作的最多次数:
- 每次操作中,将问题的正确答案改为
'T'或者'F'(也就是将answerKey[i]改为'T'或者'F')。
请你返回在不超过 k 次操作的情况下,最大 连续 'T' 或者 'F' 的数目。
示例 1:
输入: answerKey = "TTFF", k = 2
输出: 4
解释: 我们可以将两个 'F' 都变为 'T' ,得到 answerKey = "TTTT" 。
总共有四个连续的 'T' 。
示例 2:
输入: answerKey = "TFFT", k = 1
输出: 3
解释: 我们可以将最前面的 'T' 换成 'F' ,得到 answerKey = "FFFT" 。
或者,我们可以将第二个 'T' 换成 'F' ,得到 answerKey = "TFFF" 。
两种情况下,都有三个连续的 'F' 。
示例 3:
输入: answerKey = "TTFTTFTT", k = 1
输出: 5
解释: 我们可以将第一个 'F' 换成 'T' ,得到 answerKey = "TTTTTFTT" 。
或者我们可以将第二个 'F' 换成 'T' ,得到 answerKey = "TTFTTTTT" 。
两种情况下,都有五个连续的 'T' 。
思路
- 滑动窗口
- 相比于传统的滑动窗口,这道题的本质是记录当前是否有一种可以满足交换条件,记录每种的数量。判断另一种条件是否大于改变数量
代码:
class Solution {
public:
int maxConsecutiveAnswers(string answerKey, int k) {
int n = answerKey.size();
vector<int> counts(2);
int res = 0;
for(int l = 0, r = 0; r < n; r++) {
if(answerKey[r] == 'T' ) counts[0]++;
else counts[1]++;
while(l <= r && (counts[0] > k && counts[1] > k)) { // 一定要后判断,先加入后判断!
if(answerKey[l++] == 'T') counts[0]--;
else counts[1]--;
}
res = max(res, r - l + 1);
}
return res;
}
};