自学内容网 自学内容网

LeetCode 3191.使二进制数组全部等于 1 的最少操作次数 I:模拟(说是最小操作次数,其实不重复翻转就是了)

【LetMeFly】3191.使二进制数组全部等于 1 的最少操作次数 I:模拟(说是最小操作次数,其实不重复翻转就是了)

力扣题目链接:https://leetcode.cn/problems/minimum-operations-to-make-binary-array-elements-equal-to-one-i/

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

解题方法:模拟(其实是很不严格的证明)

从前到后遍历数组(遍历到倒数第三个元素),遇见 0 0 0则从当前位置开始连续翻转3个元素。

遍历结束后,若最后两个元素都是 1 1 1,则返回总翻转次数;否则则返回 − 1 -1 1

为何这样正常操作就是“最小操作次数”:

因为这样不会把“同样的三个元素”翻转多次(最小性证明),同时又不得不翻转(必要性证明)。

因为是从前向后遍历的,遇到零的话如果往前翻(前面全是1),则前面的1变成0后还需要额外次数再次翻转回1。

时空复杂度分析

  • 时间复杂度 O ( l e n ( n u m s ) ) O(len(nums)) O(len(nums))
  • 空间复杂度 O ( 1 ) O(1) O(1)

AC代码

C++
/*
011100
100100
100011
111111


011100
100100
111000
111111

0111
1001
1110
*/
class Solution {
public:
    int minOperations(vector<int>& nums) {
        int ans = 0;
        for (int i = 0; i < nums.size() - 2; i++) {
            if (!nums[i]) {
                ans++;
                // nums[i] ^= 1;  // 这个修改与否都无所谓了
                nums[i + 1] ^= 1;
                nums[i + 2] ^= 1;
            }
        }
        return nums[nums.size() - 1] & nums[nums.size() - 2] ? ans : -1;
    }
};
Go
package main

func minOperations(nums []int) int {
    ans := 0
    for i := 0; i < len(nums) - 2; i++ {
        if nums[i] == 0 {
            ans++
            nums[i + 1] ^= 1
            nums[i + 2] ^= 1
        }
    }
    if nums[len(nums) - 1] & nums[len(nums) - 2] == 1 {
        return ans
    } else {
        return -1
    }
}
Java
class Solution {
    public int minOperations(int[] nums) {
        int ans = 0;
        for (int i = 0; i < nums.length - 2; i++) {
            if (nums[i] == 0) {
                ans++;
                nums[i + 1] ^= 1;
                nums[i + 2] ^= 1;
            }
        }
        return (nums[nums.length - 1] & nums[nums.length - 2]) == 1 ? ans : -1;
    }
}
Python
from typing import List

class Solution:
    def minOperations(self, nums: List[int]) -> int:
        ans = 0
        for i in range(len(nums) - 2):
            if not nums[i]:
                ans += 1
                nums[i + 1] ^= 1
                nums[i + 2] ^= 1
        return ans if nums[-1] & nums[-2] else -1

同步发文于CSDN和我的个人博客,原创不易,转载经作者同意后请附上原文链接哦~

Tisfy:https://letmefly.blog.csdn.net/article/details/143064645


原文地址:https://blog.csdn.net/Tisfy/article/details/143064645

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!