题—2020（二分+贪心）

2020

牛客网小米选拔1 A题

题意：
给一只含有2,1,0的字符串，现在要求选出最多子序列为"2020"的个数。

思路：
二分。每次二分，判断个数是否有。
之后贪心：
假如该串要你判断是否有5个"2020"，
既然要确定有没有5个，那么串只要存在5个"2020"就行了。
串的前5个2分别为5个"2020"的第一个2.
证实这样贪心是正确的。

举个例子：
1表示第一个"2020"
3表示第三个"2020"
判断有没有四个"2020"。
串下标: 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
(该)串:2 2 2 0 0 0 2 0 2 2 0 2 0 2 0 0
每个2020位置: 1 2 3 1 2 3 4 4 1 2 1 3 2 4 3 4
这里判断：然后就是4个2020了。
代码如下：

#include<iostream>
#include<cstring>
using namespace std;
char  s[1000006];
int n;
bool check(int k){
	int a=0,b=0,c=0,d=0;
	// 记录a:2的个数   b:20的个数  c:202的个数 d:2020的个数。
	int w=0; //截止s前的k个2.
	for(int i=1;i<=n;i++){
		if(s[i]=='2'){
			if(w<k){
				w++,a++;
			}
			else if(b){
				--b;c++;
			}
		}
		else if(s[i]=='0'){
			if(a){
				--a,b++;
				}
			else 
				if(c){
				--c,d++;
			
			}
		}
	}
	//cout<<k<<endl;
	return k==d; 
}
void solve(){
	//int T;
	//cin>>T;
	while(cin>>n){
		cin>>(s+1);
		// n=strlen(s+1);
		int l=0,r=n/4;
		while(l<=r){   //二分。
			int k=(l+r)>>1;
			if(check(k)) l=k+1;
			else r=k-1;
		}
		cout<<l-1<<endl;
	}
}
int main(){
	 solve();
}