【算法题】给定一个包含 0 和 1 的数组，如果允许用 1 替换不超过“k”个 0，请找出全为 1 的最长连续子数组的长度。

最新推荐文章于 2024-08-27 12:22:52 发布

be5yond

最新推荐文章于 2024-08-27 12:22:52 发布

阅读量1.3k

点赞数

分类专栏：算法题文章标签：算法数据结构

本文链接：https://blog.csdn.net/be5yond/article/details/119919314

版权

该博客介绍了一种算法，用于找到一个数组中，允许替换最多k个0为1后，具有最长不同字符（在这里是0）的子数组的长度。算法通过滑动窗口实现，动态调整窗口大小，保持窗口内0的数量不超过k。最终返回的是最大子数组的长度。

摘要由CSDN通过智能技术生成

思路：

使用一个窗口框出子数组，判断如果窗口内0的个数小于等于k，则右边界增加扩大窗口。

若窗口内0的个数大于k，则左边界增加减小窗口内数据。

每次记录窗口size，获取最大值。

思路同【算法题】求带有 K 个不同字符的最长子字符串

代码

In [1]: def longest_sub_array_with_k_replace(array, k):
   ...:     max_len = 0
   ...:     left, right, end = 0, 0, len(array)
   ...:     zero_count = 0
   ...:
   ...:     while right <= end:
   ...:         if zero_count <= k:
   ...:             print(array[left:right])
   ...:             max_len = max(max_len, len(array[left: right]))
   ...:             try:
   ...:                 # 若新数据位置是0, zero_count +1
   ...:                 zero_count += (not array[right])
   ...:             except IndexError:
   ...:                 pass
   ...:             right += 1
   ...:         else:
   ...: