leetcode 2379. 得到 K 个黑块的最少涂色次数

南鱼座α 19 2023-03-09

题目内容

给你一个长度为 n 下标从 0 开始的字符串 blocks ,blocks[i] 要么是 ‘W’ 要么是 ‘B’ ,表示第 i 块的颜色。字符 ‘W’ 和 ‘B’ 分别表示白色和黑色。
给你一个整数 k ,表示想要 连续 黑色块的数目。
每一次操作中,你可以选择一个白色块将它 涂成 黑色块。
请你返回至少出现 一次 连续 k 个黑色块的 最少 操作次数。

示例 1:

输入:blocks = "WBBWWBBWBW", k = 7
输出:3
解释:
一种得到 7 个连续黑色块的方法是把第 0 ,3 和 4 个块涂成黑色。
得到 blocks = "BBBBBBBWBW" 。
可以证明无法用少于 3 次操作得到 7 个连续的黑块。
所以我们返回 3 。

示例 2:

输入:blocks = "WBWBBBW", k = 2
输出:0
解释:
不需要任何操作,因为已经有 2 个连续的黑块。
所以我们返回 0 。

提示:

  • n == blocks.length
  • 1 <= n <= 100
  • blocks[i] 要么是 ‘W’ ,要么是 ‘B’ 。
  • 1 <= k <= n

解题思路

滑动窗口,统计每一个窗口中’W’的个数,找出最小个数。
时间复杂度O(n)。

public static int minimumRecolors(String blocks, int k) {
        int cnt = 0;
        for (int i = 0; i < k; i++) {
            cnt += blocks.charAt(i) == 'W' ? 1 : 0;
        }
        int res = cnt;
        for (int i = k; i < blocks.length(); i++) {
            cnt += blocks.charAt(i) == 'W' ? 1 : 0;
            cnt -= blocks.charAt(i - k) == 'W' ? 1 : 0;
            res = Math.min(res, cnt);
        }
        return res;
    }

运行结果

image