题目地址:
https://www.lintcode.com/problem/digital-flip/description
给定一个 0 − 1 0-1 0−1数组,每次可以将其中一个 1 1 1变为 0 0 0或 0 0 0变为 1 1 1。要求flip若干位置使得 0 0 0的后面的数字必须是 0 0 0。问至少要flip多少个位置。
可以看出,最后要达到的效果是,使得数组先连续若干个 1 1 1,然后再若干个 0 0 0(这里的”若干“可以是 0 0 0个)。那么可以枚举分界点,如果分界点是 A [ i ] A[i] A[i]的话(也就是如果 A [ i ] A[i] A[i]是最后一个 1 1 1的话),设 A A A里一共有 x x x个 1 1 1, A [ 0 : i ] A[0:i] A[0:i]里 0 0 0的个数是 a a a, A [ i + 1 : l A − 1 ] A[i+1:l_A-1] A[i+1:lA−1]里 1 1 1的个数是 b b b,那么要flip的位置个数就是 a + b a+b a+b。同时更新最小值即可。代码如下:
public class Solution {
/**
* @param nums: the array
* @return: the minimum times to flip digit
*/
public int flipDigit(int[] nums) {
// Write your code here
// 先数一下一共多少个1
int count = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] == 1) {
count++;
}
}
// 枚举所有数都flip成0的可能
int res = count;
// 开两个变量记录A[0:i]有多少个0和1
int oneCount = 0, zeroCount = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] == 0) {
zeroCount++;
} else {
oneCount++;
}
// A[0: i]的0的个数加上A[i + 1: n - 1]的1的个数
res = Math.min(res, zeroCount + count - oneCount);
}
return res;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。