硬币(C++)

题目描述

小 B 面前的桌子上有 n 个硬币，0 表示正面，1 表示反面，只有当这 n 个硬币都是 0 朝上的时候这个他才能把这些钱收起来。现在他以一个这样的操作来翻硬币，他选择一个 x，把 x-1、x、 x+1 位置上的硬币都翻面（若x为1，则只有1，2翻面；若x为n，则只有n-1和n翻面），他现在想知道最少需要多少次操作能使所有硬币正面朝上。 如果不能使所有硬币翻面，则输出-1。

输入描述

第一行一个整数n，表示硬币数量 第二行 n 个整数，每个数字均为 0 或者 1

输出描述

输出最小操作次数

输入样例
5
0 0 1 0 0
输出样例
2
样例解释
先翻转第5枚硬币，此时正反面情况为： 0 0 1 1 1，再翻转第4枚硬币，变为0 0 0 0 0。

数据范围
对于 30% 的数据, 1≤n≤20

对于另外 20%的数据, 全部数字相等

对于 100%的数据, 1≤n≤10^6

参考代码

#include<iostream>
using namespace std;
int n, cnt, cnt1 = 1, a[1000001], b[1000001];
int main(){
	cin >> n;
	for(int i = 1; i <= n; i++){
		cin >> a[i];
		b[i] = a[i];
	}
	for(int i = 1; i < n; i++){
		if(a[i] == 0) continue;
		if(a[i] == 1) a[i+1] = !a[i+1], a[i+2] = !a[i+2], cnt++;
	}
	if(a[n] == 1) cnt = 1000000;
	
	b[1] = !b[1], b[2] = !b[2];
	for(int i = 1; i < n; i++){
		if(b[i] == 0) continue;
		if(b[i] == 1) b[i+1] = !b[i+1], b[i+2] = !b[i+2], cnt1++;
	}
	if(b[n] == 1) cnt1 = 1000000;
	if(cnt == 1000000 && cnt1 == 1000000) cout << -1;
	else cout << min(cnt, cnt1);
	
	return 0;
}