力扣3191.使二进制数全变成1

白光白光2024-10-21 23:10

给你一个二进制数组 nums 。

你可以对数组执行以下操作任意次（也可以 0 次）：

选择数组中 任意连续 3 个元素，并将它们 全部反转 。

反转一个元素指的是将它的值从 0 变 1 ，或者从 1 变 0 。

请你返回将 nums 中所有元素变为 1 的最少操作次数。如果无法全部变成 1 ，返回 -1 。

示例 1：

**输入：**nums = [0,1,1,1,0,0]

**输出：**3

解释：

我们可以执行以下操作：

选择下标为 0 ，1 和 2 的元素并反转，得到 nums = [++1++ ,++0++ ,++0++,1,0,0] 。
选择下标为 1 ，2 和 3 的元素并反转，得到 nums = [1,++1++ ,++1++ ,++0++,0,0] 。
选择下标为 3 ，4 和 5 的元素并反转，得到 nums = [1,1,1,++1++ ,++1++ ,++1++] 。

示例 2：

**输入：**nums = [0,1,1,1]

输出：-1

解释：

无法将所有元素都变为 1 。

提示：

3 <= nums.length <= 105
0 <= nums[i] <= 1

python 复制代码

class Solution:
    def minOperations(self, nums: List[int]) -> int:
        ans = 0
        for i in range(len(nums) - 2):
            if nums[i] == 0:  # 必须操作
                nums[i + 1] ^= 1
                nums[i + 2] ^= 1
                ans += 1
        return ans if nums[-2] and nums[-1] else -1

参考：灵茶山艾府

链接：https://leetcode.cn/problems/minimum-operations-to-make-binary-array-elements-equal-to-one-i/solutions/2819085/cong-zuo-dao-you-xiu-gai-pythonjavacgo-b-k38u/