力扣2024题——考试的最大困扰度

169 阅读2分钟

2024. 考试的最大困扰度

一位老师正在出一场由 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' 。

思路

  1. 滑动窗口
  2. 相比于传统的滑动窗口,这道题的本质是记录当前是否有一种可以满足交换条件,记录每种的数量。判断另一种条件是否大于改变数量

代码:

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;
    }
};