LOJ#2764. 「JOI 2013 Final」JOIOI 塔

题目地址

https://loj.ac/problem/2764

题解

真的想不到二分...不看tag的话...

考虑二分答案转化为判定问题，那么问题就变成了能不能组合出x个JOI/IOI，考虑贪心判定，倒着做，统计I的个数cnt，已组OI的个数tot，以及JOI/IOI个数ans。对于J显然直接找一个OI组成答案。对于O显然直接找I。对于I需要贪心考虑，假设目前的cnt+tot+ans>=x那么就组答案，否则做OI里面的那个I，让cnt++。（贪心考虑只需要造x个OI，造多了会浪费）

#include <bits/stdc++.h>
using namespace std;

#define ll long long
const int N = 1000010;

int n;
char s[N];

bool check(int x) {
	int ans = 0, tot = 0, cnt = 0;
	for(int i = n; i; --i) {
		if(x == ans) return 1;
		if(s[i] == 'J') {
			if(tot) --tot, ++ans;
			continue;
		}
		if(s[i] == 'O') {
			if(cnt) --cnt, ++tot;
			continue;
		}
		if(s[i] == 'I') {
			if(ans + tot + cnt < x) cnt++;
			else {
				if(tot) --tot, ++ans;
			}
		}
	}
	if(x == ans) return 1;
	return 0;
}

int main() {
	scanf("%d%s", &n, s + 1);
	int l = 0, r = n, ans = 0;
	while(l <= r) {
		int mid = (l + r) >> 1;
		if(check(mid)) ans = mid, l = mid + 1;
		else r = mid - 1;
 	}
 	printf("%d\n", ans);
}