【从零开始的LeetCode-算法】3191. 使二进制数组全部等于 1 的最少操作次数 I

九圣残炎2024-10-16 16:53

给你一个二进制数组 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

我的解答：

java 复制代码

class Solution {
    public int minOperations(int[] nums) {
        int size = nums.length;
        int res = 0;
        // 遍历除倒数最后两个以外的所有元素，因为最后两个元素不能单独反转
        for(int i = 0; i < size - 2; i++){
            // 如果当前索引元素为0，则该索引及后面两个元素都进行异或运算
            if(nums[i] == 0){
                nums[i] ^= 1;
                nums[i + 1] ^= 1;
                nums[i + 2] ^= 1;
                res++;
            }
        } 
        // 判断倒数最后两个元素是否都为1，是则返回操作次数，否则返回-1
        return nums[size - 1] + nums[size - 2] == 2 ? res : -1;
    }
}