2379. 得到 K 个黑块的最少涂色次数
难度简单
给你一个长度为 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
思路:本道题一个很直观的思路是枚举,针对所有的白色块进行枚举,我们可以将所有的白块当成一个状态为0的状态,将状态1视为黑块,然后统计变化多少块可以有k个连续黑块。
上述的方式很笨,而且最坏的情况下复杂度达到了2^100,是不可取的。试想一下,我们想要花更少的步数得到更长的连续黑块区间,应该考虑尽可能长的黑色块向两边扩展、或者是对于短的白色块区间但是可以连接两个长的黑色块区间。然而这样做编码会非常麻烦,有没有更神奇的方法呢?
其实我们没有必要非要考虑区间的起点终点,换种方式考虑问题,假设我们现在有了0-i区间上所有色块都是黑色需要的最少操作数(其实就是白色块的个数),又有了0-j(j > i)区间上所有色块都是黑色需要的最少操作数,此时我们想要知道i+1-j的块都变为黑色的最少操作数,只需要对这两个区间做减法即可,即。
此时我们如果想知道以i结束长度为k的黑块区间的操作数就是,更新所有的该值即可(即所有的位置i都作为结束计算一次操作次数)。
class Solution {
public:
int minimumRecolors(string blocks, int k) {
//用dp[i]表示0-i全部涂黑
int n = blocks.length(), plant_num[n + 5], min_num = k;
for(int idx = 0; idx < n; ++ idx){
if(!idx){
plant_num[0] = blocks[0] == 'W';
}else{
plant_num[idx] = plant_num[idx - 1] + (blocks[idx] == 'W');
}
printf("plant:%d ", plant_num[idx]);
if(idx >= k - 1){
min_num = min(min_num, idx >= k ? plant_num[idx] - plant_num[idx - k] : plant_num[idx]);
}
}
return min_num;
}
};