[codeforces 1399D] Binary String To Subsequences 交错的两个栈(0栈,1栈)

Codeforces Round #661 (Div. 3)   参与排名人数12987

[codeforces 1399D]   Binary String To Subsequences   交错的两个栈(0栈,1栈)

总目录详见https://blog.csdn.net/mrcrack/article/details/103564004

在线测评地址https://codeforces.com/contest/1399/problem/D

Problem	Lang	Verdict	Time	Memory
D - Binary String To Subsequences	GNU C++17	Accepted	62 ms	6400 KB

题目大意：给定一个只由0,1构成的字符串，将其拆解成子字符串，要求字符顺序不变，这些子串中的0,1要求间隔出现，或者只有1个1,或者只有1个0,要求分成的子字符串的个数尽可能小，输出子字符串个数，以及每个字符归属的第几个子字符串。

基本思路：有跳跃的数据，并且是10,或者01匹配的形式，很符合 栈 这种数据结构的特点。

开两个栈，一个存储0(记录当前所属的子串),另一个存储1(记录当前所属的子串),读取当前字符，若能发生相应的10配对，或者01配对，需进行相应处理，详见样例模拟(建议结合AC代码进行阅读)。

4
0011

2
1 2 2 1

位置1234
字符0011

读入字符1,s[1]='0',存入0栈,k=1
0栈数据如下
位置1
数据1

位置1234
字符0011
数值1

读入字符2,s[2]='0',存入0栈,k=2
0栈数据如下
位置1 2
数据1 2

位置1234
字符0011
数值12

读入字符2,s[3]='1',因0栈已有数据，此时发生01匹配,k=2
0栈数据如下
位置1
数据1

1栈数据如下
位置1
数据2

位置1234
字符0011
数值122

读入字符2,s[4]='1',因0栈已有数据，此时发生01匹配,k=2
0栈数据如下
位置0
数据

1栈数据如下
位置1 2
数据2 1

位置1234
字符0011
数值1221

AC代码如下：

#include 
#define maxn 200010
char s[maxn];
int stack0[maxn],top0,stack1[maxn],top1,a[maxn];
int main(){
	int t,n,i,k;
	scanf("%d",&t);
	while(t--){
		scanf("%d%s",&n,s+1);
		top0=top1=k=0;
		for(i=1;i<=n;i++){
			if(s[i]=='0'){
				if(top1>0)//10配对
					stack0[++top0]=stack1[top1],top1--;//10匹配中，1退出主导，0开始主导之后的匹配
				else k++,stack0[++top0]=k;//需开辟新的子串
				a[i]=stack0[top0];//记录i位置字母所属的子串
			}else{//s[i]=='1'
				if(top0>0)//01配对
					stack1[++top1]=stack0[top0],top0--;//01匹配中，0退出主导，1开始主导之后的匹配
				else k++,stack1[++top1]=k;//需开辟新的子串
				a[i]=stack1[top1];//记录i位置字母所属的子串
			}
		}
		printf("%d\n",k);
		printf("%d",a[1]);
		for(i=2;i<=n;i++)printf(" %d",a[i]);
		printf("\n");
	}
	return 0;
}