LeetCode 1004. 最大连续1的个数 III(双指针+滑动窗口)

题目描述

给定一个由若干 0 和 1 组成的数组 A,我们最多可以将 K 个值从 0 变成 1 。
返回仅包含 1 的最长(连续)子数组的长度。
示例 1:
输入:A = [1,1,1,0,0,0,1,1,1,1,0], K = 2
输出:6
解释:
[1,1,1,0,0,1,1,1,1,1,1]
粗体数字从 0 翻转到 1,最长的子数组长度为 6。
示例 2:
输入:A = [0,0,1,1,0,0,1,1,1,0,1,1,0,0,0,1,1,1,1], K = 3
输出:10
解释:
[0,0,1,1,1,1,1,1,1,1,1,1,0,0,0,1,1,1,1]
粗体数字从 0 翻转到 1,最长的子数组长度为 10。

思路

详见链接

代码

class Solution:
	def longestOnes(self,A,K):
		result = 0
		left = right = 0
		count = 0
		if len(A) < K:
			return K
		while right < len(A):
			if A[right] == 0:
				count += 1
			while count > K:
				if A[left] == 0:
					count -= 1
				left += 1
			result = max(result,right - left + 1)
			right += 1
		return result
test =Solution()
test.longestOnes([1,1,1,0,0,0,1,1,1,1,0],2)

效果

在这里插入图片描述

发布了38 篇原创文章 · 获赞 194 · 访问量 4705

猜你喜欢

转载自blog.csdn.net/weixin_37763870/article/details/104012477