acm-(二分、贪心)2020ICPC·小米 网络选拔赛第二场 A.2020

传送门
考虑二分答案，假设答案为$k$，那么如何取check呢？我们可以从左到右贪心的选$l_1,l_2,l_3...l_k$代表左边最近的$k$个2的位置，然后对于每个2我们都贪心地选一个右边离它最近的位置的0进行配对，这些配对的0的位置分别是$r_1,r_2,r_3...r_k$，很显然满足$l_1<l_2<...<l_k,r_1<r_2<...r_k,l_i<r_i$，然后我们再从右边到左边贪心的选$k$个离右边最近的0，按从右到左的顺序存到$r_{2k},r_{2k-1},...,r_{k+1}$，对于每个0而言同样选择左边离它最近的2，存到$l_{2k},l_{2k-1},...,l_{k+1}$中，这样子的话我们可以让$(l_i,r_i)$与$(l_{i+k},r_{i+k})$配对形成2020，如果无法形成配对，即满足$r_i\ge l_{i+k}$或$l_{1\sim k},r_{1\sim k}$与$l_{k+1\sim 2k},r_{k+1\sim 2k}$存在交集的话，说明在答案为$k$的情况下是无法配对出足够数量的$2020$的。

以上贪心显然是正确的。

char s[maxn];
int n,mi[maxn],mx[maxn],vis[maxn];

bool check(int k){
	int tot1=0,tot2=0;
	memset(vis,0,sizeof(int)*n);
	FOR(i,0,n){
		if(s[i]=='2' && tot1<k)mi[++tot1]=i;else if(s[i]=='0' && tot2<tot1)mx[++tot2]=i;
		if(tot1==k && tot2==k)break;
	}
	if(tot1!=k || tot2!=k)return false;
	tot1=0,tot2=0;
	ROF(i,n-1,0){
		if(s[i]=='0' && tot1<k)mx[(k<<1)-(tot1++)]=i;else if(s[i]=='2' && tot2<tot1)mi[(k<<1)-(tot2++)]=i;
		if(tot1==k && tot2==k)break;
	}
	if(tot1 !=k || tot2!=k)return false;
	FOR(i,1,k+1){
		if(mx[i]>=mi[i+k] || vis[mi[i]] || vis[mx[i]])return false;
		vis[mi[i+k]]=vis[mx[i+k]]=1;
	}
	return true;
}
int main(){ 
	while(~scanf("%d",&n)){
		scanf("%s",s);
		int ans=0,l=1,r=n;
		while(l<=r){
			int mid=l+r>>1;
			if(ans==mid || check(mid)){
				ans=mid;
				l=mid+1;
			}else r=mid-1;
		}
		wrn(ans);
	}
}