max_consecutive_ones.py

"""Max Consecutive Ones II"""

def max_consecutive_ones_II(bits):
  res = max_count = 0
  counter = Counter()
	for i in range(len(bits)):
		counter[bits[i]] += 1
		maxf = max(maxf, counter[bits[i]])
		if res - maxf < 1:
			res += 1
		else:
			counter[bits[i - res]] -= 1
	return res

"""Max Consecutive Ones III"""

from collections import Counter
def max_consecutive_ones_III(bits, k):
	maxf = res = 0
	counter = Counter()
	for i in range(len(bits)):
		counter[bits[i]] += 1
		maxf = max(maxf, counter[bits[i]])
		if res - maxf < k:
			res += 1
		else:
			counter[bits[i - res]] -= 1
	return res