【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
111111011100
100100
111000
1111110111
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 mainfunc minOperations(nums []int) int {ans := 0for i := 0; i < len(nums) - 2; i++ {if nums[i] == 0 {ans++nums[i + 1] ^= 1nums[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 Listclass Solution:def minOperations(self, nums: List[int]) -> int:ans = 0for i in range(len(nums) - 2):if not nums[i]:ans += 1nums[i + 1] ^= 1nums[i + 2] ^= 1return ans if nums[-1] & nums[-2] else -1
同步发文于CSDN和我的个人博客,原创不易,转载经作者同意后请附上原文链接哦~
Tisfy:https://letmefly.blog.csdn.net/article/details/143064645