LeetCode: 2024. 考试的最大困扰度

99 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第11天,点击查看活动详情

2024. 考试的最大困扰度

来源:力扣(LeetCode)

链接: leetcode-cn.com/problems/ma…

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

提示:

n == answerKey.length 1 <= n <= 51045 * 10^4 answerKey[i] 要么是 'T' ,要么是 'F' 1 <= k <= n

解法

  • 滑动窗的变体:只要求最大连续指定字符的数目时,本题和「1004. 最大连续1的个数 III」完全一致。**在指定字符的情况下,可以计算其最大连续数目。**具体地,使用滑动窗口的方法,从左到右枚举右端点,维护区间中另一种字符的数量为 sum,当sum 超过 k时候,需要一直让左端点右移,直到sumksum≤k。移动过程中,记录滑动窗口的最大长度,即为指定字符的最大连续数目。

    本题的答案为分别指定字符为 T 和 F 时的最大连续数目二者的较大值。

代码实现

  • python实现
class Solution:
    def maxConsecutiveAnswers(self, answerKey: str, k: int) -> int:
        def helper(ch: str) -> int:
            ans, left, diff_sum = 0, 0, 0
            for right in range(len(answerKey)):
                diff_sum += answerKey[right] != ch
                while diff_sum > k:  # 这个地方是while循环,一直移动到小于等于k为止
                    diff_sum -= answerKey[left] != ch  # left左移动之后,如果不等于目标元素,说明移动到不匹配的那个位置,就能减去1
                    left += 1
                ans = max(ans, right-left+1)
            return ans
        return max(helper('T'), helper('F'))
  • c++实现
class Solution {
public:
    int helper(string answerKey, char ch, int k)
    {
        int ans = 0;
        int left = 0;
        int diff_sum = 0;
        int size = answerKey.length();
        for(int right=0; right<size; right++)
        {
            if(answerKey[right] != ch)
            {
                diff_sum++;
            }
            while(diff_sum > k)
            {
                if(answerKey[left] != ch)
                {
                    diff_sum--;
                }
                left++;
            }
            ans = max(ans, right-left+1);
        }
        return ans;
    }

    int maxConsecutiveAnswers(string answerKey, int k) {
        return max(helper(answerKey, 'T', k), helper(answerKey, 'F', k));
    }
};

复杂度分析

  • 时间复杂度: O(n)O(n)
  • 空间复杂度: O(1)O(1)