给定一个二进制数组 nums ,如果最多可以翻转一个 0 ,则返回数组中连续 1 的最大个数。
示例 1:
输入:nums = [1,0,1,1,0]
输出:4
解释:翻转第一个 0 可以得到最长的连续 1。
当翻转以后,最大连续 1 的个数为 4。
示例 2:
输入:nums = [1,0,1,1,0,1]
输出:4
提示:
1 <= nums.length <= 105
nums[i] 不是 0 就是 1.
方法1
class Solution:
def findMaxConsecutiveOnes(self, nums: List[int]) -> int:
win_dic={1:0,0:0}
max_len=-1
right,left=0,0
n=len(nums)
while right<n:
key= nums[right]
win_dic[key]+=1
if win_dic[0]>1:
win_dic[nums[left]]-=1
left+=1
if win_dic[0] == 1:
max_len = max(max_len, win_dic[1]+1)
else:
max_len = max(max_len, win_dic[1])
right+=1
return max_len
方法2
class Solution:
def findMaxConsecutiveOnes(self, nums: List[int]) -> int:
res, prev, cnt = 0, 0, 0
for num in nums:
print("num:",num)
cnt += 1
if num == 0:
prev = cnt
cnt = 0
res = max(res, prev + cnt)
return res
在这段代码中,变量 cnt 用于记录当前连续的 1 的个数。
具体来说,代码通过遍历二进制数组 nums 的每个元素,当遇到元素为 1 时,将 cnt 值加一,表示当前连续的 1 的个数增加了。如果遇到元素为 0,意味着需要翻转一个 0,所以要更新 prev 和 cnt。
cnt:记录当前连续的 1 的个数,包括了已经翻转过的 0 后面的连续 1 的个数。当遇到 0 时,需要重新开始计数。
prev:记录之前已经翻转过的 0 前面的连续 1 的个数。当遇到新的 0 时,需要将 prev 更新为当前的 cnt 值,以保存之前的连续 1。
在遍历过程中,通过不断更新 res 的值,来记录连续 1 的最大个数。每次遇到新的 0,就计算当前连续 1 的个数(包括已翻转的 0 和当前的连续 1),然后与之前的最大值进行比较,更新 res。
最后,返回 res,即为连续 1 的最大个数(包括翻转一个 0 后的情况)。
注意:代码中使用了 max() 函数来比较和更新最大值,确保得到最大的连续 1 的个数。