995. K 连续位的最小翻转次数
在仅包含 0 和 1 的数组 A 中,一次 K 位翻转包括选择一个长度为 K 的(连续)子数组,同时将子数组中的每个 0 更改为 1,而每个 1 更改为 0。
返回所需的 K 位翻转的最小次数,以便数组没有值为 0 的元素。如果不可能,返回 -1。
示例 1:
输入:A = [0,1,0], K = 1
输出:2
解释:先翻转 A[0],然后翻转 A[2]。
示例 2:
输入:A = [1,1,0], K = 2
输出:-1
解释:无论我们怎样翻转大小为 2 的子数组,我们都不能使数组变为 [1,1,1]。
示例 3:
输入:A = [0,0,0,1,0,1,1,0], K = 3
输出:3
解释:
翻转 A[0],A[1],A[2]: A变成 [1,1,1,1,0,1,1,0]
翻转 A[4],A[5],A[6]: A变成 [1,1,1,1,1,0,0,0]
翻转 A[5],A[6],A[7]: A变成 [1,1,1,1,1,1,1,1]
提示:
1 <= A.length <= 30000
1 <= K <= A.length
解题:滑动窗口
因为模拟所有翻转可能的耗费太大,所以我们通过数学手段简化,用奇偶来表示翻转的结果
用队列que存储翻转子数组的头部
1.que.append(i)必要条件:i元素需要被翻转(len(que) % 2 == A[i]
)
2.que.popleft()必要条件:子数组无法覆盖元素i(i >= que[0] + K
)
判断元素i是否要被翻转条件:len(que) % 2 == A[i]
(奇偶——元素)
“不可能”条件:i + K > N
(超长)
class Solution:
def minKBitFlips(self, A: List[int], K: int) -> int:
N = len(A)
que = collections.deque()
res = 0
for i in range(N):
if que and i >= que[0] + K:
que.popleft()
if len(que) % 2 == A[i]:
if i + K > N: return -1
que.append(i)
res += 1
return res