Submission #1666434

---

Source Code Expand

#include <bits/stdc++.h>

using namespace std;

const int MX = 500000;

int solve(vector<int>& a) {
	if (a.empty()) return 0;
	
	int f = 0, g = -2 * MX, h = -2 * MX;
	for (int x : a) {
		int ff = max(max(f, h), g + x);
		int gg = max(f, g + x - 1);
		if (x > 1) gg = max(gg, h);
		int hh = max(f, max(g, h - 1)) + x;
		
		f = ff;
		g = gg;
		h = hh;
	}
	
	return max(f, g);
}

char s[MX + 2];

int main() {
	int n;
	ignore = scanf("%d %s", &n, s);
	s[n] = s[n + 1] = '0';
	n += 2;
	
	int ans = 0;
	vector<int> a;
	for (int i = 0, o = 0, z = 0; i < n; i++) {
		if (s[i] == '1') {
			o++;
			z = 0;
		}
		else {
			if (o != 0) {
				a.push_back(o);
				o = 0;
			}
			z++;
			if (z == 2) {
				ans += solve(a);
				a.clear();
				o = z = 0;
			}
		}
	}
	
	printf("%d\n", ans);
	
	return 0;
}

Submission Info

Judge Result